NPKit: enable reduce scatter profiling (#1580)
This commit is contained in:
gecommit door
GitHub
bovenliggende
063c6cfc11
commit
f957c4fe22
@@ -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
|
||||
|
||||
|
||||
Verwijs in nieuw issue
Block a user