diff --git a/src/graph/rome_models.cc b/src/graph/rome_models.cc index 68cd10ee99..c90677f332 100644 --- a/src/graph/rome_models.cc +++ b/src/graph/rome_models.cc @@ -294,6 +294,17 @@ static struct rcclRomeModel rome_model_43 = { .ringBase = "0 1 2 3 4 5 6 7|0 2 5 7 4 6 1 3|0 3 1 6 4 7 5 2|0 7 6 5 4 3 2 1", }; +static struct rcclRomeModel rome_model_55 = { + .nGpus = 8, .nCpus = 4, .nNics = 0, .nLinks = 3, + .gpuIds = { 0x100000, 0x200000, 0x300000, 0x400000, 0x500000, 0x600000, 0x700000, 0x800000, }, + .nicIds = { }, + .gpuNuma = { 0, 0, 1, 1, 2, 2, 3, 3, }, + .nicNuma = { }, + .connMatrix = { 0, 1, 1, 1, 0, 0, 0, 0, 1, 0, 1, 1, 0, 0, 0, 0, 1, 1, 0, 1, 0, 0, 0, 0, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 0, 0, 1, 0, 1, 1, 0, 0, 0, 0, 1, 1, 0, 1, 0, 0, 0, 0, 1, 1, 1, 0, }, + .pattern = "20202020", + .ringBase = "0 1 2 3 4 5 6 7|7 6 5 4 3 2 1 0|2 3 0 1 6 7 4 5|5 4 7 6 1 0 3 2", +}; + static struct rcclRomeModel romeTopoModels[] = { rome_model_22, rome_model_25, @@ -318,6 +329,7 @@ static struct rcclRomeModel romeTopoModels[] = { rome_model_52, rome_model_53, rome_model_43, + rome_model_55, }; /* Parse user defined rings. Format is like : diff --git a/tools/scripts/topo_val.sh b/tools/scripts/topo_val.sh index 0f05a9fe56..ad2a5d4eff 100755 --- a/tools/scripts/topo_val.sh +++ b/tools/scripts/topo_val.sh @@ -21,7 +21,7 @@ DIR="$(cd -P "$(dirname "${BASH_SOURCE[0]}")" && pwd)" -for i in {0..54} +for i in {0..55} do if [[ $i -eq 50 ]] || [[ $i -eq 51 ]] then diff --git a/tools/topo_expl/models/topo_8p_rome_vm1.xml b/tools/topo_expl/models/topo_8p_rome_vm1.xml new file mode 100644 index 0000000000..9f3bcc938e --- /dev/null +++ b/tools/topo_expl/models/topo_8p_rome_vm1.xml @@ -0,0 +1,69 @@ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + diff --git a/tools/topo_expl/topo_expl.cpp b/tools/topo_expl/topo_expl.cpp index 622949b2df..46675d6219 100644 --- a/tools/topo_expl/topo_expl.cpp +++ b/tools/topo_expl/topo_expl.cpp @@ -124,6 +124,7 @@ NodeModelDesc model_descs[] = { {1, "topo_8p_90a.xml", "single node gfx90a"}, {4, "topo_8p_rome_4n_1.xml", "4 nodes node 8 gfx908 Rome 4 NICs NPS=4"}, {4, "topo_collnet_n4.xml", "4 nodes collnet 4 NICs for multiple SAT"}, + {1, "topo_8p_rome_vm1.xml", "single node 8 gfx908 Rome VM"}, }; int main(int argc,char* argv[])