Added new unit tests for AllReduce with Bias API (#2036)

* Added new unit tests for AllReduce with Bias API * Address review comments
2025-12-03 17:37:34 -06:00
parent 185e78a8f0
commit 7c12b0b76b
9 changed files with 586 additions and 18 deletions
@@ -718,6 +718,11 @@ namespace RcclUnitTesting
                                                    &numOutputElements);
          optionalArgs.redOp = redOps[rdIdx];
          optionalArgs.root = roots[rtIdx] % this->numActiveRanks;
+          // Set biasNumElements if bias is enabled
+          if (optionalArgs.useBias)
+          {
+            optionalArgs.biasNumElements = numOutputElements;
+          }
          this->SetCollectiveArgs(funcTypes[ftIdx],
                                  dataTypes[dtIdx],
                                  numInputElements,