Skip to content

Commit fc713c6

Browse files
authored
Merge pull request #2527 from devitocodes/amd-base-again
CI: update amd base to working ucx/rocm versions
2 parents 6331ebf + 4214228 commit fc713c6

1 file changed

Lines changed: 4 additions & 12 deletions

File tree

docker/Dockerfile.amd

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,8 @@ ARG ROCM_VERSION=6.3.2
77

88
FROM rocm/dev-ubuntu-22.04:${ROCM_VERSION}-complete as sdk-base
99

10-
ARG UCX_BRANCH="v1.14.1"
11-
ARG OMPI_BRANCH="v5.0.6"
10+
ARG UCX_BRANCH="v1.16.0"
11+
ARG OMPI_BRANCH="v5.0.x"
1212

1313
# Update and Install basic Linux development tools
1414
RUN rm /etc/apt/sources.list.d/* \
@@ -51,16 +51,6 @@ RUN ln -s /opt/rocm/llvm/bin/offload-arch /opt/rocm/bin/offload-arch | echo "off
5151
# Install tmpi
5252
RUN curl https://raw.githubusercontent.com/Azrael3000/tmpi/master/tmpi -o /usr/local/bin/tmpi
5353

54-
# Adding rocm/cmake to the Environment
55-
ENV PATH=$ROCM_HOME/bin:$ROCM_HOME/profiler/bin:$ROCM_HOME/opencl/bin:/opt/cmake/bin:$PATH \
56-
LD_LIBRARY_PATH=$ROCM_HOME/lib:$ROCM_HOME/lib64:$ROCM_HOME/llvm/lib:$LD_LIBRARY_PATH \
57-
C_INCLUDE_PATH=$ROCM_HOME/include:$C_INCLUDE_PATH \
58-
CPLUS_INCLUDE_PATH=$ROCM_HOME/include:$CPLUS_INCLUDE_PATH \
59-
CPATH=$ROCM_HOME/include:$CPATH \
60-
INCLUDE=$ROCM_HOME/include:$INCLUDE
61-
62-
WORKDIR /tmp
63-
6454
# Install UCX
6555
RUN cd /tmp/ \
6656
&& git clone --depth 1 https://github.com/openucx/ucx.git -b ${UCX_BRANCH} \
@@ -100,6 +90,8 @@ RUN cd /tmp \
10090
--enable-mpi1-compatibility \
10191
--enable-mpi-fortran=no \
10292
--disable-debug \
93+
--disable-oshmem \
94+
--disable-man-pages \
10395
&& make -j ${nproc} \
10496
&& make install
10597

0 commit comments

Comments
 (0)