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

* Added new unit tests for AllReduce with Bias API

* Address review comments
This commit is contained in:
Atul Kulkarni
2025-12-03 17:37:34 -06:00
committed by GitHub
orang tua 185e78a8f0
melakukan 7c12b0b76b
9 mengubah file dengan 586 tambahan dan 18 penghapusan
+5
Melihat File
@@ -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,