Tweak clique channel usage for gfx908 (#374)

Этот коммит содержится в:
gilbertlee-amd
2021-05-21 15:36:21 -06:00
коммит произвёл GitHub
родитель 50da1b48af
Коммит 8e817ecd6d
+4 -2
Просмотреть файл
@@ -367,8 +367,10 @@ ncclResult_t CliqueManager::GetNumChannelsToUse(ncclFunc_t const coll,
else numChannels = 2;
break;
case 908:
if (totalBytes <= 262144) numChannels = 4;
else numChannels = 14;
if (totalBytes <= 131072) numChannels = 2;
else if (totalBytes <= 524288) numChannels = 6;
else if (totalBytes <= 1048576) numChannels = 13;
else numChannels = 16;
break;
case 910:
if (totalBytes <= 262144) numChannels = 4;