Dockerfile: Update rccl version and fix issue in rocm5.1.1 dockerfile (#354)

**Description**
Update rccl version and fix issue in rocm5.1.1 dockerfile.
This commit is contained in:
Yuting Jiang 2022-05-27 10:46:40 +08:00 коммит произвёл GitHub
Родитель 81a4146bc1
Коммит e08b6d3a1c
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 4 добавлений и 4 удалений

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

@ -110,13 +110,13 @@ RUN cd /tmp && \
cd /tmp && \
rm -rf mlc
# Install rccl with commitid 700b473
# Install rccl with commitid 6707a27
RUN cd /tmp && \
git clone https://github.com/ROCmSoftwarePlatform/rccl.git && \
cd rccl && git checkout 700b473 && \
cd rccl && git checkout 6707a27 && \
mkdir build && cd build && \
CXX=/opt/rocm/bin/hipcc cmake -DCMAKE_INSTALL_PREFIX=/opt/rocm/rccl .. && \
make -j ${NUM_MAKE_JOBS} && make install && \
CXX=/opt/rocm/bin/hipcc cmake -DCMAKE_INSTALL_PREFIX=/usr/local .. && \
make && make install && \
cd /tmp && \
rm -rf rccl