From 4d20b4b758a9502394be7435fa6726a3fb034625 Mon Sep 17 00:00:00 2001
From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com>
Date: Tue, 25 Jul 2023 11:15:21 -0700
Subject: [PATCH] Replace atomicExch with __atomic_store_n (#818)

* Replace atomicExch with __atomic_store_n

* Remove extra semicolon
---
 src/collectives/device/common.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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); }