Tweak clique channel usage for gfx908 (#374)

[ROCm/rccl commit: 8e817ecd6d]
This commit is contained in:
gilbertlee-amd
2021-05-21 15:36:21 -06:00
کامیت شده توسط GitHub
والد 92bcdcf5b0
کامیت ea17b05518
@@ -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;