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

* Add option to enable multiple SAT in SHARP

* Extend number of NICs to 16

[ROCm/rccl commit: 961922ea02]
Этот коммит содержится в:
Wenkai Du authored and GitHub committed 2021-06-03 19:45:18 -07:00
1 родитель fd94c55afe
Коммит 4b31e521e9
5 files changed
+36 -47

No files matched your search

+4 -1
Просмотреть файл
@@ -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