added npkit support into the all_gather run ring algorithm (#790)

[ROCm/rccl commit: 9bba4a2f2a]
Этот коммит содержится в:
akolliasAMD
2023-06-29 13:59:54 -06:00
коммит произвёл GitHub
родитель 3d153f7583
Коммит e1ac484d4e
3 изменённых файлов: 78 добавлений и 0 удалений
+10
Просмотреть файл
@@ -100,4 +100,14 @@
#define NPKIT_PRIM_COLLECT_DATA_PROCESS_TIME 0x4F
#define NPKIT_EVENT_ALL_GATHER_RING_ENTRY 0x50
#define NPKIT_EVENT_ALL_GATHER_RING_EXIT 0x51
#define NPKIT_EVENT_ALL_GATHER_RING_SEND_ENTRY 0x52
#define NPKIT_EVENT_ALL_GATHER_RING_SEND_EXIT 0x53
#define NPKIT_EVENT_ALL_GATHER_RING_RECV_COPY_SEND_ENTRY 0x54
#define NPKIT_EVENT_ALL_GATHER_RING_RECV_COPY_SEND_EXIT 0x55
#define NPKIT_EVENT_ALL_GATHER_RING_DIRECT_RECV_ENTRY 0x56
#define NPKIT_EVENT_ALL_GATHER_RING_DIRECT_RECV_EXIT 0x57
#endif