added atomic acquire for gfx12 on prims_simple (#1382)

[ROCm/rccl commit: af5678641d]
This commit is contained in:
akolliasAMD
2024-10-18 11:26:38 -06:00
committad av GitHub
förälder a5cc8edd9b
incheckning ad2c8c3eb8
+4
Visa fil
@@ -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 <int DirectRecv, int DirectSend, int Recv, int Send, int Src, int Dst>