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]
This commit is contained in:
Wenkai Du authored and GitHub committed 2021-06-03 19:45:18 -07:00
1 vanhempi fd94c55afe
commit 4b31e521e9
5 files changed
+36 -47

No files matched your search

@@ -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