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

[ROCm/rccl commit: 0f14e5a640]
2023-07-10 09:31:49 -07:00
parent 9e8cc15c61
commit 3ce8711153
4 changed files with 39 additions and 1 deletions
@@ -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
 };