Merge remote-tracking branch 'nccl/master' into develop

Этот коммит содержится в:
Wenkai Du
2021-04-30 16:57:36 -07:00
родитель 4f8e788a61 a46ea10583
Коммит a4ea1fed5b
80 изменённых файлов: 3136 добавлений и 1675 удалений
+3
Просмотреть файл
@@ -191,6 +191,9 @@ int main(int argc,char* argv[])
assert(node_model!=0);
comm[i].topo = node_model->getSystem(i);
bootstrapAllGather(&comm[i], allGather1Data);
// Mark channels as non initialized.
for (int c=0; c<MAXCHANNELS; c++) comm[i].channels[c].id = -1;
NCCLCHECK(ncclCalloc((uint32_t**)&comm[i].p2pNet, 1));
}
struct ncclTopoGraph *treeGraph, *ringGraph, *collNetGraph;