microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-11 08:50:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

updates to expand worldsize

Browse Source
This commit is contained in:
Ekow Wellington
2026-04-27 23:45:08 -05:00
parent 719e9124af
commit fa52c565e6
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/mscclpp/__main__.py
View File

@@ -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",

2
src/ext/collectives/algorithm_collection_builder.cc
View File

@@ -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) {