AI/rocm-systems
2
0
Forka 0
Codice Problemi Pull Request Actions Pacchetti Progetti Rilasci Wiki Attività

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

Sfoglia il codice sorgente
This commit is contained in:
BertanDogancay
2025-04-23 20:46:36 -07:00
parent 82afb2bcfe dcdc67c40b
commit a6bf9bfc9e
118 ha cambiato i file con 12789 aggiunte e 3358 eliminazioni
Scarica il file Patch Scarica il file Diff Expand all files Collapse all files
test/common/CollectiveArgs.cpp
+12 -12
Vedi File
@@ -195,18 +195,18 @@ namespace RcclUnitTesting
scalarsPerRank.Attach(scalarsPerRank.ptr);
switch (this->dataType)
{
case ncclInt8: ss << scalarsPerRank.I1[this->globalRank]; break;
case ncclUint8: ss << scalarsPerRank.U1[this->globalRank]; break;
case ncclInt32: ss << scalarsPerRank.I4[this->globalRank]; break;
case ncclUint32: ss << scalarsPerRank.U4[this->globalRank]; break;
case ncclInt64: ss << scalarsPerRank.I8[this->globalRank]; break;
case ncclUint64: ss << scalarsPerRank.U8[this->globalRank]; break;
case ncclFp8E4M3: ss << scalarsPerRank.F1[this->globalRank]; break;
case ncclFloat32: ss << scalarsPerRank.F4[this->globalRank]; break;
case ncclFloat64: ss << scalarsPerRank.F8[this->globalRank]; break;
case ncclFp8E5M2: ss << scalarsPerRank.B1[this->globalRank]; break;
case ncclBfloat16: ss << scalarsPerRank.B2[this->globalRank]; break;
default: ss << "(UNKNOWN)";
case ncclInt8: ss << scalarsPerRank.I1[this->globalRank]; break;
case ncclUint8: ss << scalarsPerRank.U1[this->globalRank]; break;
case ncclInt32: ss << scalarsPerRank.I4[this->globalRank]; break;
case ncclUint32: ss << scalarsPerRank.U4[this->globalRank]; break;
case ncclInt64: ss << scalarsPerRank.I8[this->globalRank]; break;
case ncclUint64: ss << scalarsPerRank.U8[this->globalRank]; break;
case ncclFloat8e4m3: ss << scalarsPerRank.F1[this->globalRank]; break;
case ncclFloat32: ss << scalarsPerRank.F4[this->globalRank]; break;
case ncclFloat64: ss << scalarsPerRank.F8[this->globalRank]; break;
case ncclFloat8e5m2: ss << scalarsPerRank.B1[this->globalRank]; break;
case ncclBfloat16: ss << scalarsPerRank.B2[this->globalRank]; break;
default: ss << "(UNKNOWN)";
}
ss << " ";
}