File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -7,8 +7,8 @@ ARG ROCM_VERSION=6.3.2
77
88FROM 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
1414RUN 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
5252RUN 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
6555RUN 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
You can’t perform that action at this time.
0 commit comments