(cherry picked from commit 94da229a7788d74685d1591a4e75a8341de64f41)
This commit is contained in:
@@ -21,7 +21,7 @@
|
||||
|
||||
DIR="$(cd -P "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
|
||||
for i in {0..78}
|
||||
for i in {0..80}
|
||||
do
|
||||
if [[ $i -eq 50 ]] || [[ $i -eq 51 ]]
|
||||
then
|
||||
|
||||
@@ -151,6 +151,8 @@ NodeModelDesc model_descs[] = {
|
||||
{2, "topo_8p1h_4.xml", "2 nodes 8P1H Alt."},
|
||||
{1, "topo_8p1h_5.xml", "Single node 8P1H Alt."},
|
||||
{2, "topo_8p1h_5.xml", "2 nodes 8P1H Alt."},
|
||||
{1, "topo_8p_940.xml", "Single node gfx940 8P"},
|
||||
{2, "topo_4p_940.xml", "2 nodes gfx940 4P"},
|
||||
};
|
||||
|
||||
NCCL_PARAM(MaxCTAs, "MAX_CTAS", MAXCHANNELS);
|
||||
|
||||
Reference in New Issue
Block a user