NPKit update (#844)

* NPKit update 1. Enable NPKit for MSCCL kernels 2. Fix NPKit context index calculation for sendrecv kernels * Update build script for npkit [ROCm/rccl commit: d33a70e620]
2023-08-09 08:30:40 +08:00
parent cb291157f4
commit 18811f6159
7 changed files with 95 additions and 2 deletions
@@ -112,5 +112,9 @@
 #define NPKIT_EVENT_NET_TEST_ENTRY                              0x58
 #define NPKIT_EVENT_NET_TEST_EXIT                               0x59

+#define NPKIT_EVENT_MSCCL_GENERIC_OP_ENTRY                      0x5A
+#define NPKIT_EVENT_MSCCL_GENERIC_OP_EXIT                       0x5B
+#define NPKIT_EVENT_MSCCL_REDUCE_ENTRY                          0x5C
+#define NPKIT_EVENT_MSCCL_REDUCE_EXIT                           0x5D

 #endif