collnet: support multiple NICs (#335)

Dieser Commit ist enthalten in:
Wenkai Du
2021-03-25 20:59:32 -07:00
committet von GitHub
Ursprung 69c0b44553
Commit d87dc7c2e8
10 geänderte Dateien mit 209 neuen und 36 gelöschten Zeilen
+2
Datei anzeigen
@@ -36,6 +36,8 @@ struct allGather3Data_t{
struct ncclTopoRanks topoRanks;
};
void initCollNet();
ncclResult_t ncclTopoGetSystem(const char* xmlTopoFile, struct ncclTopoSystem** system);
ncclResult_t ncclTopoGetSystemFromXml(struct ncclXml* xml, struct ncclTopoSystem** topoSystem);