AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

NPKit: enable reduce scatter profiling (#1580)

Browse Source 
[ROCm/rccl commit: f957c4fe22]
This commit is contained in:
Wenkai Du
2025-03-04 10:03:56 -08:00
committed by GitHub
parent 751370bb70
commit 086fa823db
3 changed files with 90 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/install.sh
+8
View File
@@ -377,6 +377,14 @@ if [[ "${npkit_enabled}" == true ]]; then
-DENABLE_NPKIT_EVENT_MSCCL_INIT_EXIT \
-DENABLE_NPKIT_EVENT_BROADCAST_RING_ENTRY \
-DENABLE_NPKIT_EVENT_BROADCAST_RING_EXIT \
-DENABLE_NPKIT_EVENT_REDUCE_SCATTER_RING_ENTRY \
-DENABLE_NPKIT_EVENT_REDUCE_SCATTER_RING_EXIT \
-DENABLE_NPKIT_EVENT_REDUCE_SCATTER_RING_SEND_ENTRY \
-DENABLE_NPKIT_EVENT_REDUCE_SCATTER_RING_SEND_EXIT \
-DENABLE_NPKIT_EVENT_REDUCE_SCATTER_RING_RECV_REDUCE_SEND_ENTRY \
-DENABLE_NPKIT_EVENT_REDUCE_SCATTER_RING_RECV_REDUCE_SEND_EXIT \
-DENABLE_NPKIT_EVENT_REDUCE_SCATTER_RING_RECV_REDUCE_COPY_ENTRY \
-DENABLE_NPKIT_EVENT_REDUCE_SCATTER_RING_RECV_REDUCE_COPY_EXIT \
-DENABLE_NPKIT_PRIM_COLLECT_DATA_PROCESS_TIME"
fi