Fix size matching in MSCCL (#1318)

[ROCm/rccl commit: 7830af5844]
This commit is contained in:
Ziyue Yang authored and GitHub committed 2024-10-01 13:32:41 -07:00
1 parent efc500d2ff
commit cf980e9b9c
1 file changed
+1 -1
@@ -330,7 +330,7 @@ static ncclResult_t mscclInternalSchedulerSelectAlgo(int rank, struct mscclSched
auto &m = status.algoMetas[i];
size_t nBytes = param->count * ncclTypeSize(param->dataType) * m.sizeMultiplier;
bool msgSizeIsValid =
param->count > 0 && (param->count % m.nChunksPerLoop) == 0 &&
param->count > 0 && ((param->count * m.sizeMultiplier) % m.nChunksPerLoop) == 0 &&
nBytes >= m.minBytes && (m.maxBytes == 0 || nBytes <= m.maxBytes);
if (msgSizeIsValid &&
m.nRanks == param->nRanks &&