From c2064adcc7c26b5cd56477dc71c417f1418465c7 Mon Sep 17 00:00:00 2001
From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com>
Date: Tue, 8 Jun 2021 13:58:20 -0700
Subject: [PATCH] Add support for another Rome model (#385)

---
 src/graph/rome_models.cc | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/graph/rome_models.cc b/src/graph/rome_models.cc
index 37a324b638..68cd10ee99 100755
--- a/src/graph/rome_models.cc
+++ b/src/graph/rome_models.cc
@@ -283,6 +283,17 @@ static struct rcclRomeModel rome_model_53 = {
   .ringBase = "N0 0 1 2 3 4 5 6 7 N3|N3 7 6 5 4 3 2 1 0 N0|N1 2 3 0 1 6 7 4 5 N2|N2 5 4 7 6 1 0 3 2 N1",
 };
 
+static struct rcclRomeModel rome_model_43 = {
+  .nGpus = 8, .nCpus = 4, .nNics = 0, .nLinks = 3,
+  .gpuIds = { 0x63000, 0x43000, 0x27000, 0x3000, 0xe3000, 0xc3000, 0xa3000, 0x83000, },
+  .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|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 romeTopoModels[] = {
   rome_model_22,
   rome_model_25,
@@ -306,6 +317,7 @@ static struct rcclRomeModel romeTopoModels[] = {
   rome_model_49,
   rome_model_52,
   rome_model_53,
+  rome_model_43,
 };
 
 /* Parse user defined rings. Format is like :