added atomic acquire for gfx12 on prims_simple (#1382)
[ROCm/rccl commit: af5678641d]
This commit is contained in:
committad av
GitHub
förälder
a5cc8edd9b
incheckning
ad2c8c3eb8
@@ -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>
|
||||
|
||||
Referens i nytt ärende
Block a user