From 87423792fd5ed999dfa44ea0bef5fdd2325f8eaa Mon Sep 17 00:00:00 2001 From: gilbertlee-amd <44450918+gilbertlee-amd@users.noreply.github.com> Date: Tue, 18 May 2021 15:36:07 -0600 Subject: [PATCH] Tune clique-based AllReduce for device type 908 (#372) * Changing switch-over point for clique-based AllReduce [ROCm/rccl commit: ddceadc3133174826eca7e3b46abb2ae928533e2] --- projects/rccl/src/clique/CliqueManager.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/projects/rccl/src/clique/CliqueManager.cc b/projects/rccl/src/clique/CliqueManager.cc index ee38baf860..06a350f969 100644 --- a/projects/rccl/src/clique/CliqueManager.cc +++ b/projects/rccl/src/clique/CliqueManager.cc @@ -261,7 +261,7 @@ void CliqueManager::SetByteLimits() switch (m_gcnArch) { case 906: m_allReduceByteLimit = 16777216; break; - case 908: m_allReduceByteLimit = 536870912; break; + case 908: m_allReduceByteLimit = 8388608; break; default: m_allReduceByteLimit = 16777216; break; } }