Tune clique-based AllReduce for device type 908 (#372)
* Changing switch-over point for clique-based AllReduce
Tá an tiomantas seo le fáil i:
tiomanta ag
GitHub
tuismitheoir
2daadcc834
tiomantas
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;
|
||||
}
|
||||
}
|
||||
|
||||
Tagairt in Eagrán Nua
Cuir bac ar úsáideoir