Add Feature - Add NPKit Support in RCCL (#564)

* apply npkit * fix bug * add npkit in readme
2022-06-21 05:30:19 +08:00
parent f274c865c1
commit 6e93fafdc3
14 changed files with 1236 additions and 8 deletions
@@ -58,6 +58,10 @@ struct ncclProxySubArgs {
  uint64_t end;
  void* requests[NCCL_STEPS];
  void* profilingEvents[NCCL_STEPS];
+
+#if defined(ENABLE_NPKIT) && defined(ENABLE_NPKIT_EVENT_NET_SEND_ENTRY) && defined(ENABLE_NPKIT_EVENT_NET_SEND_EXIT)
+  int npKitSizesFifo[NCCL_STEPS];
+#endif
 };

 struct ncclProxyArgs {