* minor tuning for tree ll
Этот коммит содержится в:
akolliasAMD
2022-08-03 15:07:44 -06:00
коммит произвёл GitHub
родитель 9089e68a99
Коммит 4cecdc9be5
+3 -3
Просмотреть файл
@@ -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 {