Merge remote-tracking branch 'nccl/master' into develop

This commit is contained in:
BertanDogancay committed 2025-04-23 20:47:43 -07:00
کامیت a6bf9bfc9e
118 files changed
+12789 -3358

No files matched your search

+3 -1
مشاهده پرونده
@@ -223,7 +223,9 @@ namespace RcclUnitTesting
NCCLCHECK(ncclGroupStart());
for (int rank = 0; rank < numRanks; rank++)
NCCLCHECK(ncclAllReduce(gpuInput[rank], gpuOutput[rank], N, ncclInt, ncclSum, comms[rank], stream[rank]));
NCCLCHECK(ncclGroupEnd());
ncclResult_t res = ncclGroupEnd();
if (res != ncclSuccess) continue;
const auto start = Clock::now();