Implement atomicswap for CUDA

vchuravy · vchuravy · commit 0c9756ff39bd · 2025-05-13T11:30:55.000+02:00
diff --git a/ext/AtomixCUDAExt.jl b/ext/AtomixCUDAExt.jl
@@ -48,6 +48,8 @@ end
             CUDA.atomic_min!(ptr, x)
         elseif op === max
             CUDA.atomic_max!(ptr, x)
+        elseif op === Atomix.right
+            CUDA.atomic_xchg!(ptr, x)
         else
             error("not implemented")
         end
