Tune clique-based AllReduce for device type 908 (#372)
* Changing switch-over point for clique-based AllReduce
Этот коммит содержится в:
коммит произвёл
GitHub
родитель
2daadcc834
Коммит
ddceadc313
@@ -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;
|
||||
}
|
||||
}
|
||||
|
||||
Ссылка в новой задаче
Block a user