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

Bu işleme şunda yer alıyor:
Marzieh Berenjkoub
2026-01-20 13:01:49 -06:00
işleme 858b4e76eb
240 değiştirilmiş dosya ile 16266 ekleme ve 3578 silme
+6 -6
Dosyayı Görüntüle
@@ -647,18 +647,18 @@ namespace RcclUnitTesting
this->streams[groupId][localRank][collArg.streamIdx]),
"ncclScatter");
break;
case ncclCollAllToAll:
CHILD_NCCL_CALL_RANK(errCode, ncclAllToAll(
case ncclCollAlltoAll:
CHILD_NCCL_CALL_RANK(errCode, ncclAlltoAll(
collArg.inputGpu.ptr,
collArg.outputGpu.ptr,
collArg.numInputElements / collArg.totalRanks,
collArg.dataType,
this->comms[localRank],
this->streams[groupId][localRank][collArg.streamIdx]),
"ncclAllToAll");
"ncclAlltoAll");
break;
case ncclCollAllToAllv:
CHILD_NCCL_CALL_RANK(errCode, ncclAllToAllv(
case ncclCollAlltoAllv:
CHILD_NCCL_CALL_RANK(errCode, ncclAlltoAllv(
collArg.inputGpu.ptr,
collArg.options.sendcounts + (this->rankOffset + localRank)*this->totalRanks,
collArg.options.sdispls + (this->rankOffset + localRank)*this->totalRanks,
@@ -668,7 +668,7 @@ namespace RcclUnitTesting
collArg.dataType,
this->comms[localRank],
this->streams[groupId][localRank][collArg.streamIdx]),
"ncclAllToAllv");
"ncclAlltoAllv");
break;
case ncclCollSend:
CHILD_NCCL_CALL_RANK(errCode, ncclSend(