move fence earlier for vmapnt

JuliaSIMD · May 11, 2024 · b413ba8 · b413ba8
1 parent eeaa0b2
commit b413ba8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/simdfunctionals/map.jl b/src/simdfunctionals/map.jl
@@ -182,8 +182,8 @@ function (m::VmapClosure{NonTemporal,F,D,N,A})(
   (offset, stop) = ThreadingUtilities.load(p, Int, offset)
 
   _vmap_singlethread!(m.f, dest, start, stop, Val{NonTemporal}(), args)
-  ThreadingUtilities._atomic_store!(p, ThreadingUtilities.SPIN)
   NonTemporal && Threads.atomic_fence()
+  ThreadingUtilities._atomic_store!(p, ThreadingUtilities.SPIN)
   nothing
 end