Add another Rome model (#497)

This commit is contained in:
Wenkai Du
2022-02-12 10:30:16 -08:00
zatwierdzone przez GitHub
rodzic f3c2cafd9d
commit 02096c9936
5 zmienionych plików z 117 dodań i 2 usunięć
+3
Wyświetl plik
@@ -811,6 +811,9 @@ ncclResult_t ncclTopoCompute(ncclTopoSystem* system, struct ncclTopoGraph* graph
if (graph->nChannels) return ncclSuccess;
// try to match 1H16P
NCCLCHECK(parse1H16P(system, graph));
if (graph->nChannels) return ncclSuccess;
// try to match 4H4P
NCCLCHECK(parse4H4P(system, graph));
}
if (graph->nChannels) return ncclSuccess;