Give preference to path with more XGMI connections

[ROCm/rccl commit: b3c9852634]
This commit is contained in:
Wenkai Du
2020-05-14 15:27:49 -07:00
parent d1cd60fbf4
commit 27519fd019
5 changed files with 117 additions and 7 deletions
@@ -76,6 +76,7 @@ const char *model_descriptions[] = {
"4 nodes with 8 VEGA20 GPUs XGMI 4P2H 1 NIC",
"4 nodes with 8 VEGA20 GPUs XGMI 4P2H 1 NIC 2nd Hive",
"4 nodes with 8 VEGA20 GPUs XGMI 4P2H 2 NIC",
"single node 8 VEGA20 Rome",
NULL,
};
@@ -166,6 +167,10 @@ int main(int argc,char* argv[])
network.AddNode(node);
}
break;
case 13:
node = new NodeModel("topo_8p_rome.xml");
network.AddNode(node);
break;
default:
printf("Invalid model_id %d\n", model_id);
exit(0);