@@ -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 {
|
||||
|
||||
Ссылка в новой задаче
Block a user