diff --git a/src/collectives/device/common.h b/src/collectives/device/common.h
index f645e97ae2..18cf382652 100644
--- a/src/collectives/device/common.h
+++ b/src/collectives/device/common.h
@@ -22,7 +22,7 @@
 
 #ifdef __GFX9__
 #define STORE(DST, SRC) \
-  { __threadfence(); atomicExch((unsigned long long *)(DST), (SRC)); }
+  { __threadfence(); __atomic_store_n((DST), (SRC), __ATOMIC_RELAXED); }
 #else
 #define STORE(DST, SRC) \
   { __atomic_store_n((DST), (SRC), __ATOMIC_SEQ_CST); }