Print KL/CL/KE events for all warps (#1544)

* Print KL/CL/KE events for all warps

* Fix count off-by-one issue

* Fix opCount in KE and restore CPU thread option

* Simplify count calculation

[ROCm/rccl commit: ebf7e2305e]
This commit is contained in:
Wenkai Du
2025-02-12 13:36:31 -08:00
کامیت شده توسط GitHub
والد 7fdbcdfdec
کامیت 170cc1afde
3فایلهای تغییر یافته به همراه30 افزوده شده و 15 حذف شده
@@ -419,7 +419,8 @@ struct ncclCollTrace {
uint8_t type;
uint8_t bid;
int16_t funcIndex;
uint32_t data_0;
uint32_t data_0:24;
uint8_t tid;
uint8_t channelId;
uint64_t timeStamp:56;
union {