Tweak clique channel usage for gfx908 (#374)

Ta zmiana jest zawarta w:
gilbertlee-amd authored and GitHub committed 2021-05-21 15:36:21 -06:00
1 rodzic 50da1b48af
złożenie 8e817ecd6d
Plik 1 zmieniony
+4 -2
+4 -2
Wyświetl plik
@@ -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;