diff --git a/src/enqueue.cc b/src/enqueue.cc index 2dfaa1303b..12281cb402 100644 --- a/src/enqueue.cc +++ b/src/enqueue.cc @@ -464,11 +464,11 @@ static ncclResult_t getAlgoInfo(struct ncclInfo* info, int collNetTypeSupport, i info->nChannels = nc; if (!userTuneInput) { // always respect user settings - if (info->nBytes <= 196608) { + if (info->nBytes <= 900000) { info->protocol = NCCL_PROTO_LL; info->algorithm = NCCL_ALGO_TREE; - info->nChannels = std::min(comm->nChannels, info->nBytes <= 65536? 4 : 12); - } else if (info->nBytes <= 1048576) { + info->nChannels = std::min(comm->nChannels, info->nBytes <= 16384? 4 : 24); + } else if (info->nBytes <= 2200008) { info->protocol = NCCL_PROTO_LL; info->algorithm = NCCL_ALGO_RING; } else {