npkit: separate network timing between send and test (#798)

src/include/npkit/npkit_event.h

@@ -109,5 +109,8 @@
 #define NPKIT_EVENT_ALL_GATHER_RING_DIRECT_RECV_ENTRY           0x56
 #define NPKIT_EVENT_ALL_GATHER_RING_DIRECT_RECV_EXIT            0x57
 
+#define NPKIT_EVENT_NET_TEST_ENTRY                              0x58
+#define NPKIT_EVENT_NET_TEST_EXIT                               0x59
+
 
 #endif

src/include/proxy.h

@@ -74,6 +74,7 @@ struct ncclProxySubArgs {
 
 #if defined(ENABLE_NPKIT) && defined(ENABLE_NPKIT_EVENT_NET_SEND_ENTRY) && defined(ENABLE_NPKIT_EVENT_NET_SEND_EXIT)
   int npKitSizesFifo[NCCL_STEPS];
+  uint64_t timestamp[NCCL_STEPS];
 #endif
 };