AI/rocm-systems
2
0
Förgrening 0
Kod Ärenden Pull-förfrågningar Actions Packages Projekt Släpp Wiki Aktiviteter

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

Bläddra i källkod 
[ROCm/rccl commit: 9bba4a2f2a]
This commit is contained in:
akolliasAMD
2023-06-29 13:59:54 -06:00
committad av GitHub
förälder 3d153f7583
incheckning e1ac484d4e
3 ändrade filer med 78 tillägg och 0 borttagningar
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/src/include/npkit/npkit_event.h
+10
Visa fil
@@ -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