improving kernel traces on opCount bits and adding channelId in ncclCollTrace (#1485)

[ROCm/rccl commit: 69d0134ed2]
Tento commit je obsažen v:
Dingming Wu
2025-01-10 07:57:46 -08:00
odevzdal GitHub
rodič b9e7e3024b
revize b6ae2fd71d
3 změnil soubory, kde provedl 9 přidání a 7 odebrání
+2 -1
Zobrazit soubor
@@ -46,7 +46,8 @@
uint32_t pos = atomicAdd(&ncclShmem.collTraceTail->tail, 1)%COLLTRACE_NUM_ITEMS; \
struct ncclCollTrace* collTrace = ncclShmem.collTrace+pos; \
collTrace->timeStamp = wall_clock64(); \
collTrace->bid = blockIdx.x;
collTrace->bid = blockIdx.x; \
collTrace->channelId = ncclShmem.channelId;
// TODO: switch to atomicInc after llvm crash is fixed
// uint32_t pos = atomicInc(&ncclShmem.collTraceTail->tail, COLLTRACE_NUM_ITEMS)