From 8e817ecd6df13803a669c0010367569855bc3501 Mon Sep 17 00:00:00 2001
From: gilbertlee-amd <44450918+gilbertlee-amd@users.noreply.github.com>
Date: Fri, 21 May 2021 15:36:21 -0600
Subject: [PATCH] Tweak clique channel usage for gfx908 (#374)

---
 src/clique/CliqueManager.cc | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/clique/CliqueManager.cc b/src/clique/CliqueManager.cc
index 06a350f969..7870ff9bef 100644
--- a/src/clique/CliqueManager.cc
+++ b/src/clique/CliqueManager.cc
@@ -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;