updates to expand worldsize

2026-05-12 01:10:22 +00:00 · 2026-04-27 23:45:08 -05:00
parent 719e9124af
commit fa52c565e6
2 changed files with 3 additions and 3 deletions
--- 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",