NPKit: enable reduce scatter profiling (#1580)

This commit is contained in:
Wenkai Du
2025-03-04 10:03:56 -08:00
gecommit door GitHub
bovenliggende 063c6cfc11
commit f957c4fe22
3 gewijzigde bestanden met toevoegingen van 90 en 1 verwijderingen
+8
Bestand weergeven
@@ -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