Enable collnet in RCCL (#333)
* Enable CollNet and use different number of channels
* topo_expl: enable collnet
[ROCm/rccl commit: 1d6244b18d]
이 커밋은 다음에 포함됨:
1 부모
260b54adb9
커밋
287ed0f18a
14 파일이 변경됨
+393
-24
검색과 일치하는 파일 없음
@@ -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
|
||||
새 이슈에서 참조
사용자 차단