diff --git a/apps/nccl/src/nccl.cu b/apps/nccl/src/nccl.cu index a414ffe85..2bd2a4422 100644 --- a/apps/nccl/src/nccl.cu +++ b/apps/nccl/src/nccl.cu @@ -216,7 +216,7 @@ static ncclResult_t ncclAllReduceFallback(const void* sendbuff, void* recvbuff, mscclpp::DeviceHandle* smOutChannels = nullptr; // Creating the channels - if (count * ncclTypeSize(datatype) <= comm->largeMessageSizeBoundary) { + if (count * ncclTypeSize(datatype) <= (1 << 20)) { auto sendIt = comm->channelScratchInfos.find(sendKey); if (sendIt == comm->channelScratchInfos.end()) { std::vector channels =