AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Tweak clique channel usage for gfx908 (#374)

Browse Source
This commit is contained in:
gilbertlee-amd
2021-05-21 15:36:21 -06:00
committed by GitHub
parent 50da1b48af
commit 8e817ecd6d
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/clique/CliqueManager.cc
+4 -2
View File
@@ -367,8 +367,10 @@ ncclResult_t CliqueManager::GetNumChannelsToUse(ncclFunc_t const coll,
else numChannels = 2; else numChannels = 2;
break; break;
case 908: case 908:
if (totalBytes <= 262144) numChannels = 4; if (totalBytes <= 131072) numChannels = 2;
else numChannels = 14; else if (totalBytes <= 524288) numChannels = 6;
else if (totalBytes <= 1048576) numChannels = 13;
else numChannels = 16;
break; break;
case 910: case 910:
if (totalBytes <= 262144) numChannels = 4; if (totalBytes <= 262144) numChannels = 4;