[ROCm/rccl commit: 0e5f4d0662]
이 커밋은 다음에 포함됨:
@@ -47,19 +47,6 @@ class NpKit {
|
||||
}
|
||||
}
|
||||
|
||||
static inline __device__ void CollectGpuEventLDS(uint8_t type, int64_t size, uint32_t rsvd, uint64_t timestamp) {
|
||||
#if defined(ENABLE_NPKIT)
|
||||
if (ncclShmem.event_buffer_head < LDS_NUM_EVENTS) {
|
||||
NpKitEvent& event = ncclShmem.event_buffer[ncclShmem.event_buffer_head];
|
||||
event.fields.type = type;
|
||||
event.fields.size = size < 0 ? 0 : size;
|
||||
event.fields.rsvd = rsvd;
|
||||
event.fields.timestamp = timestamp;
|
||||
ncclShmem.event_buffer_head++;
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
static void CollectCpuEvent(uint8_t type, int64_t size, uint32_t rsvd, uint64_t timestamp, int channel_id);
|
||||
|
||||
static uint64_t *GetCpuTimestamp();
|
||||
|
||||
새 이슈에서 참조
사용자 차단