reverting the (Reduce NPKit latency overhead in MSCCL kernel) PR #893 (#1525)

[ROCm/rccl commit: 0e5f4d0662]
이 커밋은 다음에 포함됨:
Pedram Alizadeh
2025-02-14 11:03:43 -05:00
커밋한 사람 GitHub
부모 30eecfdb25
커밋 37f46f1669
3개의 변경된 파일15개의 추가작업 그리고 41개의 파일을 삭제
-13
파일 보기
@@ -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();