npkit: add broadcast trace (#1166)

2024-05-07 14:00:16 -07:00
parent 92a4fc6204
commit a0cef69110
3 changed files with 17 additions and 0 deletions
@@ -340,6 +340,8 @@ if ($npkit_enabled); then
    -DENABLE_NPKIT_EVENT_MSCCL_RECV_REDUCE_COPY_EXIT \
    -DENABLE_NPKIT_EVENT_MSCCL_INIT_ENTRY \
    -DENABLE_NPKIT_EVENT_MSCCL_INIT_EXIT \
+    -DENABLE_NPKIT_EVENT_BROADCAST_RING_ENTRY \
+    -DENABLE_NPKIT_EVENT_BROADCAST_RING_EXIT \
    -DENABLE_NPKIT_PRIM_COLLECT_DATA_PROCESS_TIME"
 fi