msccl: add NPKIT profiling for MSCCL send-recv
Cette révision appartient à :
@@ -322,6 +322,12 @@ if ($npkit_enabled); then
|
||||
-DENABLE_NPKIT_EVENT_MSCCL_GENERIC_OP_EXIT \
|
||||
-DENABLE_NPKIT_EVENT_MSCCL_REDUCE_ENTRY \
|
||||
-DENABLE_NPKIT_EVENT_MSCCL_REDUCE_EXIT \
|
||||
-DENABLE_NPKIT_EVENT_MSCCL_SEND_ENTRY \
|
||||
-DENABLE_NPKIT_EVENT_MSCCL_SEND_EXIT \
|
||||
-DENABLE_NPKIT_EVENT_MSCCL_RECV_ENTRY \
|
||||
-DENABLE_NPKIT_EVENT_MSCCL_RECV_EXIT \
|
||||
-DENABLE_NPKIT_EVENT_MSCCL_RUN_ENTRY \
|
||||
-DENABLE_NPKIT_EVENT_MSCCL_RUN_EXIT \
|
||||
-DENABLE_NPKIT_PRIM_COLLECT_DATA_PROCESS_TIME"
|
||||
fi
|
||||
|
||||
|
||||
Référencer dans un nouveau ticket
Bloquer un utilisateur