diff --git a/projects/rccl/src/device/prims_simple.h b/projects/rccl/src/device/prims_simple.h index 588d2f1b1b..bb9b02bc41 100644 --- a/projects/rccl/src/device/prims_simple.h +++ b/projects/rccl/src/device/prims_simple.h @@ -101,7 +101,11 @@ private: #endif // volatile is faster than acquire but not as correct. Make sure reduceCopy // loads data using volatile so it doesn't see stale data in L1. +#if defined(__gfx1200__) || defined(__gfx1201__) + return __atomic_load_n(ptr, __ATOMIC_ACQUIRE); +#else return __atomic_load_n(ptr, __ATOMIC_RELAXED); +#endif } template