AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[topo_expl] Adding -n option to override number of nodes (#1134)

Browse Source 
[ROCm/rccl commit: 93982533d7]
This commit is contained in:
gilbertlee-amd
2024-04-04 15:11:47 -06:00
committed by GitHub
parent 890fafc2f7
commit 62b9f0d3a7
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/tools/topo_expl/topo_expl.cpp
+8 -2
View File
@@ -171,7 +171,7 @@ int main(int argc,char* argv[])
int maxCTAsEnv;
if (!cmdOptionExists(argv, argv + argc, "-m")) {
printf("Usage: ./topo_expl -m model_id\n");
printf("Usage: ./topo_expl -m model_id [-n numNodes=1]\n");
printf("List of model_id:\n");
for (int i = 0; i < num_models; i++)
printf(" %d: %s\n", i, model_descs[i].description);
@@ -194,7 +194,13 @@ int main(int argc,char* argv[])
initCollNet();
NodeModelDesc *desc = &model_descs[model_id];
for (int i=0; i<desc->num_nodes; i++) {
int numNodes = desc->num_nodes;
if (cmdOptionExists(argv, argv + argc, "-n")) {
char *numNodesStr = getCmdOption(argv, argv + argc, "-n");
if (numNodesStr)
numNodes = atol(numNodesStr);
}
for (int i=0; i < numNodes; i++) {
node = new NodeModel(desc->filename);
network.AddNode(node);
}