Update RCCL in 5.1.x Dockerfile

Update RCCL to https://github.com/wenkaidu/rccl/tree/nic_map.
This commit is contained in:
Yifan Xiong 2022-07-01 11:15:03 +08:00
Родитель 620192a242
Коммит 8548b1ad08
1 изменённых файлов: 3 добавлений и 3 удалений

Просмотреть файл

@ -113,10 +113,10 @@ RUN cd /tmp && \
cp ./Linux/mlc /usr/local/bin/ && \
rm -rf ./Linux mlc.tgz
# Install rccl with commitid 6707a27
# Install rccl
RUN cd /tmp && \
git clone https://github.com/ROCmSoftwarePlatform/rccl.git && \
cd rccl && git checkout 6707a27 && \
git clone https://github.com/wenkaidu/rccl.git && \
cd rccl && git checkout nic_map && \
mkdir build && cd build && \
CXX=/opt/rocm/bin/hipcc cmake -DCMAKE_INSTALL_PREFIX=/usr/local .. && \
make && make install && \