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

* 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 [ROCm/rccl commit: 4237caad69]
2025-03-11 09:32:09 -07:00
parent a79fa36b77
commit afd04a5117
3 changed files with 4 additions and 2 deletions
@@ -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,