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

Αυτή η υποβολή περιλαμβάνεται σε:
Wenkai Du
2021-04-30 16:57:36 -07:00
γονέας 4f8e788a61 a46ea10583
υποβολή a4ea1fed5b
80 αρχεία άλλαξαν με 3136 προσθήκες και 1675 διαγραφές
@@ -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;