Add option to enable multiple SAT in SHARP (#380)

* Add option to enable multiple SAT in SHARP

* Extend number of NICs to 16
This commit is contained in:
Wenkai Du
2021-06-03 19:45:18 -07:00
committato da GitHub
parent 903c84050d
commit 961922ea02
5 ha cambiato i file con 36 aggiunte e 47 eliminazioni
+4 -1
Vedi File
@@ -21,11 +21,14 @@
DIR="$(cd -P "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
for i in {0..53}
for i in {0..54}
do
if [[ $i -eq 50 ]] || [[ $i -eq 51 ]]
then
NCCL_COLLNET_ENABLE=1 $DIR/../topo_expl/topo_expl -m $i > "topo_m$i.log"
elif [[ $i -eq 54 ]]
then
RCCL_ENABLE_MULTIPLE_SAT=1 NCCL_COLLNET_ENABLE=1 $DIR/../topo_expl/topo_expl -m $i > "topo_m$i.log"
else
$DIR/../topo_expl/topo_expl -m $i > "topo_m$i.log"
fi