AI/rocm-systems
2
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

msccl: add NPKIT profiling for MSCCL send-recv

Browse Source 
[ROCm/rccl commit: a283f55f12]
This commit is contained in:
Nusrat Islam committed 2023-09-08 13:11:16 -05:00
1 parent ecd3fb42b0
commit ffbfe43500
3 files changed
+58 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

projects/rccl/src/include/npkit/npkit_event.h
+6
View File
@@ -116,5 +116,11 @@
#define NPKIT_EVENT_MSCCL_GENERIC_OP_EXIT 0x5B
#define NPKIT_EVENT_MSCCL_REDUCE_ENTRY 0x5C
#define NPKIT_EVENT_MSCCL_REDUCE_EXIT 0x5D
#define NPKIT_EVENT_MSCCL_SEND_ENTRY 0x5E
#define NPKIT_EVENT_MSCCL_SEND_EXIT 0x5F
#define NPKIT_EVENT_MSCCL_RECV_ENTRY 0x60
#define NPKIT_EVENT_MSCCL_RECV_EXIT 0x61
#define NPKIT_EVENT_MSCCL_RUN_ENTRY 0x62
#define NPKIT_EVENT_MSCCL_RUN_EXIT 0x63
#endif