modifying PATTERN_TREE case

ROCm · Feb 11, 2025 · f366914 · f366914
1 parent 73a0f8e
commit f366914
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/src/graph/rome_models.cc b/src/graph/rome_models.cc
@@ -2092,7 +2092,7 @@ ncclResult_t parseRome4P2H(struct ncclTopoSystem* system, struct ncclTopoGraph*
       }
     }
     break;
-  case NCCL_TOPO_PATTERN_BALANCED_TREE:
+  case NCCL_TOPO_PATTERN_BALANCED_TREE: case NCCL_TOPO_PATTERN_TREE:
     if (romeTopoModels[i].treeBase != nullptr) {
       NCCLCHECK(parseGraphLight(romeTopoModels[i].treeBase, system, graph, g));
       if (graph->nChannels) return ncclSuccess;
@@ -2118,12 +2118,6 @@ ncclResult_t parseRome4P2H(struct ncclTopoSystem* system, struct ncclTopoGraph*
       }
     }
     break;
-  case NCCL_TOPO_PATTERN_TREE:
-    if (romeTopoModels[i].treeBase != nullptr) {
-      NCCLCHECK(parseGraphLight(romeTopoModels[i].treeBase, system, graph, g));
-      if (graph->nChannels) return ncclSuccess;
-    }
-    break;
   }
   return ncclSuccess;
 }