microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-12 01:10:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NPKIT] Adding the NPKIT support for kernel allreduce7 in mscclpp-nccl (#399)

Browse Source
This commit is contained in:
Pedram Alizadeh
2025-01-03 15:38:57 -05:00
committed by GitHub
parent ba0d0d68b8
commit 97eaca2bd2
5 changed files with 79 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
include/mscclpp/npkit/npkit_event.hpp
View File

@@ -37,7 +37,10 @@
#define NPKIT_EVENT_EXECUTOR_INIT_ENTRY 0x19
#define NPKIT_EVENT_EXECUTOR_INIT_EXIT 0x1A
#define NPKIT_EVENT_EXECUTOR_OP_BASE_ENTRY 0x1B
#define NPKIT_EVENT_EXECUTOR_OP_BASE_EXIT 0x2E
#define NPKIT_EVENT_KERNEL_ALLREDUCE_ENTRY 0x1B
#define NPKIT_EVENT_KERNEL_ALLREDUCE_EXIT 0x1C
#define NPKIT_EVENT_EXECUTOR_OP_BASE_ENTRY 0x1D
#define NPKIT_EVENT_EXECUTOR_OP_BASE_EXIT 0x30
#endif