Enable collnet in RCCL (#333)

* Enable CollNet and use different number of channels

* topo_expl: enable collnet

[ROCm/rccl commit: 1d6244b18d]
이 커밋은 다음에 포함됨:
Wenkai Du 작성하고 GitHub가 2021-03-19 12:58:13 -07:00를 커밋함
1 부모 260b54adb9
커밋 287ed0f18a
14 파일이 변경됨
+393 -24

검색과 일치하는 파일 없음

+7 -2
파일 보기
@@ -21,8 +21,13 @@
DIR="$(cd -P "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
for i in {0..49}
for i in {0..50}
do
$DIR/../topo_expl/topo_expl -m $i > "topo_m$i.log"
if [[ $i -eq 50 ]]
then
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
$DIR/../TopoVisual/topo_visual.sh -i "topo_m$i.log"
done