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

Limit P2P channels per peer to not exceeding max channels (#1594)

Browse Source 
* Limit P2P channels per peer to not exceeding max channels

* [UT] test single GPU cases for all collectives

* [UT] fix out of range root value
This commit is contained in:
Wenkai Du
2025-03-11 09:32:09 -07:00
committed by GitHub
parent f6c6d451a9
commit 4237caad69
3 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/common/EnvVars.cpp
+1 -1
View File
@@ -204,7 +204,7 @@ namespace RcclUnitTesting
getArchInfo(&isGfx12, "gfx12");
showNames = GetEnvVar("UT_SHOW_NAMES" , 1);
minGpus = GetEnvVar("UT_MIN_GPUS" , 2);
minGpus = GetEnvVar("UT_MIN_GPUS" , 1);
maxGpus = GetEnvVar("UT_MAX_GPUS" , numDetectedGpus);
processMask = GetEnvVar("UT_PROCESS_MASK", UT_SINGLE_PROCESS | UT_MULTI_PROCESS);
verbose = GetEnvVar("UT_VERBOSE" , 0);
test/common/TestBed.cpp
+1 -1
View File
@@ -695,7 +695,7 @@ namespace RcclUnitTesting
&numInputElements,
&numOutputElements);
optionalArgs.redOp = redOps[rdIdx];
optionalArgs.root = roots[rtIdx];
optionalArgs.root = roots[rtIdx] % this->numActiveRanks;
this->SetCollectiveArgs(funcTypes[ftIdx],
dataTypes[dtIdx],
numInputElements,