From fa52c565e6f049088f2473c2ad1b013f62d3a34d Mon Sep 17 00:00:00 2001
From: Ekow Wellington <t-ekoww@microsoft.com>
Date: Mon, 27 Apr 2026 23:45:08 -0500
Subject: [PATCH] updates to expand worldsize

---
 python/mscclpp/__main__.py                          | 4 ++--
 src/ext/collectives/algorithm_collection_builder.cc | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/python/mscclpp/__main__.py b/python/mscclpp/__main__.py
index cab500ce..64f83d67 100644
--- a/python/mscclpp/__main__.py
+++ b/python/mscclpp/__main__.py
@@ -58,9 +58,9 @@ default_algo_configs = [
         "function": def_algo.allgather_mrc,
         "spec": AlgoSpec(
             name="allgather_mrc",
-            collective=AllGather(2, 1, True),
+            collective=AllGather(4, 1, True),
             nranks_per_node=1,
-            world_size=2,
+            world_size=4,
             in_place=True,
             instances=1,
             protocol="Simple",
diff --git a/src/ext/collectives/algorithm_collection_builder.cc b/src/ext/collectives/algorithm_collection_builder.cc
index 1ea4adb0..fd9c9d63 100644
--- a/src/ext/collectives/algorithm_collection_builder.cc
+++ b/src/ext/collectives/algorithm_collection_builder.cc
@@ -114,7 +114,7 @@ AlgorithmCollection AlgorithmCollectionBuilder::buildDefaultDslAlgorithms(int ra
   static const std::vector<DslAlgoConfig> defaultAlgoConfigs = {
       {"allreduce_2nodes_1K_64K.json", "allreduce", 8, 16, {{"default", 1}}},
       {"allreduce_2nodes_64K_2M.json", "allreduce", 8, 16, {{"default", 1}}},
-      {"allgather_mrc.json", "allgather", 1, 2, {{"default", 1}}}};
+      {"allgather_mrc.json", "allgather", 1, 4, {{"default", 1}}}};
   AlgorithmCollection collection;
 
   static auto generateFileId = [](const std::string& input) {