Replace atomicAdd with _hip_atmoc_fetch_add in getting colltrace tail position (#1539)

Этот коммит содержится в:
Dingming Wu
2025-02-10 08:53:25 -08:00
коммит произвёл GitHub
родитель 6cb0599e38
Коммит e8fb1335fd
+1 -1
Просмотреть файл
@@ -43,7 +43,7 @@
#endif
#ifdef ENABLE_COLLTRACE
#define INC_COLL_TRACE \
uint32_t pos = atomicAdd(&ncclShmem.collTraceTail->tail, 1)%COLLTRACE_NUM_ITEMS; \
uint32_t pos = __hip_atomic_fetch_add(&ncclShmem.collTraceTail->tail, 1, __ATOMIC_SEQ_CST, __HIP_MEMORY_SCOPE_WORKGROUP)%COLLTRACE_NUM_ITEMS; \
struct ncclCollTrace* collTrace = ncclShmem.collTrace+pos; \
collTrace->timeStamp = wall_clock64(); \
collTrace->bid = blockIdx.x; \