From 0c9756ff39bde98abcd169b386c44aef2444c98c Mon Sep 17 00:00:00 2001 From: Valentin Churavy Date: Tue, 13 May 2025 10:52:56 +0200 Subject: [PATCH] Implement atomicswap for CUDA --- ext/AtomixCUDAExt.jl | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ext/AtomixCUDAExt.jl b/ext/AtomixCUDAExt.jl index 5a79d62..fc92895 100644 --- 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