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

* apply npkit

* fix bug

* add npkit in readme
Этот коммит содержится в:
Ziyue Yang
2022-06-21 05:30:19 +08:00
коммит произвёл GitHub
родитель f274c865c1
Коммит 6e93fafdc3
14 изменённых файлов: 1236 добавлений и 8 удалений
+4
Просмотреть файл
@@ -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 {