Add 8P6L multi-node models (#239)

[ROCm/rccl commit: d5f90e19b5]
Αυτή η υποβολή περιλαμβάνεται σε:
Wenkai Du authored and GitHub committed 2020-07-21 14:10:36 -07:00
1 γονέας 56d8c7c893
υποβολή f604fc774e
15 files changed
+830 -32

No files matched your search

@@ -77,6 +77,12 @@ const char *model_descriptions[] = {
"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",
"4 nodes gfx908 8P6L 1 NIC",
"4 nodes gfx908 8P6L 2 NICs",
"4 nodes gfx908 8P6L 3 NICs",
"4 nodes gfx908 8P6L 4 NICs",
"4 nodes gfx908 8P6L 5 NICs",
"4 nodes gfx908 8P6L 6 NICs",
NULL,
};
@@ -171,6 +177,48 @@ int main(int argc,char* argv[])
node = new NodeModel("topo_8p_rome.xml");
network.AddNode(node);
break;
case 14:
for (int i=0; i<4; i++) {
node = new NodeModel("topo_8p6l.xml");
network.AddNode(node);
}
break;
case 15:
for (int i=0; i<4; i++) {
node = new NodeModel("topo_8p6l_1nic.xml");
network.AddNode(node);
}
break;
case 16:
for (int i=0; i<4; i++) {
node = new NodeModel("topo_8p6l_2nic.xml");
network.AddNode(node);
}
break;
case 17:
for (int i=0; i<4; i++) {
node = new NodeModel("topo_8p6l_3nic.xml");
network.AddNode(node);
}
break;
case 18:
for (int i=0; i<4; i++) {
node = new NodeModel("topo_8p6l_4nic.xml");
network.AddNode(node);
}
break;
case 19:
for (int i=0; i<4; i++) {
node = new NodeModel("topo_8p6l_5nic.xml");
network.AddNode(node);
}
break;
case 20:
for (int i=0; i<4; i++) {
node = new NodeModel("topo_8p6l_6nic.xml");
network.AddNode(node);
}
break;
default:
printf("Invalid model_id %d\n", model_id);
exit(0);