collnet: support multiple NICs (#335)

Цей коміт міститься в:
Wenkai Du authored and GitHub committed 2021-03-25 20:59:32 -07:00
1 джерело 69c0b44553
коміт d87dc7c2e8
10 files changed
+209 -36

No files matched your search

+2
Переглянути файл
@@ -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);