added atomic acquire for gfx12 on prims_simple (#1382)

[ROCm/rccl commit: af5678641d]
이 커밋은 다음에 포함됨:
akolliasAMD
2024-10-18 11:26:38 -06:00
커밋한 사람 GitHub
부모 a5cc8edd9b
커밋 ad2c8c3eb8
+4
파일 보기
@@ -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>