Initial support for clique-based kernels (#276)

* Initial support for clique-based kernels
Этот коммит содержится в:
gilbertlee-amd
2020-10-13 11:22:04 -06:00
коммит произвёл GitHub
родитель ae008fd2db
Коммит 2b8184808d
27 изменённых файлов: 1969 добавлений и 7 удалений
+1 -1
Просмотреть файл
@@ -12,7 +12,7 @@
ncclResult_t bootstrapNetInit();
ncclResult_t bootstrapCreateRoot(ncclUniqueId* commId, bool idFromEnv);
ncclResult_t bootstrapGetUniqueId(ncclUniqueId* out);
ncclResult_t bootstrapInit(ncclUniqueId* id, int rank, int nranks, void** commState);
ncclResult_t bootstrapInit(ncclUniqueId* id, int rank, int nranks, void** commState, int* rootPid); // [RCCL] Adding rootPid
ncclResult_t bootstrapAllGather(void* commState, void* allData, int size);
ncclResult_t bootstrapSend(void* commState, int peer, void* data, int size);
ncclResult_t bootstrapRecv(void* commState, int peer, void* data, int size);