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

[ROCm/rccl commit: 69d0134ed2]
Bu işleme şunda yer alıyor:
Dingming Wu
2025-01-10 07:57:46 -08:00
işlemeyi yapan: GitHub
ebeveyn b9e7e3024b
işleme b6ae2fd71d
3 değiştirilmiş dosya ile 9 ekleme ve 7 silme
+2 -1
Dosyayı Görüntüle
@@ -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)