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