* 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]
Cette révision appartient à :
Ziyue Yang
2023-08-09 08:30:40 +08:00
révisé par GitHub
Parent cb291157f4
révision 18811f6159
7 fichiers modifiés avec 95 ajouts et 2 suppressions
+4
Voir le fichier
@@ -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