Implement atomicswap for CUDA

vchuravy · vchuravy · commit bb2e06699e9c · 2025-05-13T10:52:56.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
