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

NPKit: enable reduce scatter profiling (#1580)

Bläddra i källkod
This commit is contained in:
Wenkai Du
2025-03-04 10:03:56 -08:00
committad av GitHub
förälder 063c6cfc11
incheckning f957c4fe22
3 ändrade filer med 90 tillägg och 1 borttagningar
Download Patch File Download Diff File Expand all files Collapse all files
install.sh
+8
Visa fil
@@ -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