Skip to content

Commit

Permalink
WIP use dev open MPI
Browse files Browse the repository at this point in the history
  • Loading branch information
streeve committed May 14, 2024
1 parent 1673b24 commit 2342dd2
Showing 1 changed file with 1 addition and 18 deletions.
19 changes: 1 addition & 18 deletions docker/Dockerfile.hipcc
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y \
python-dev \
autotools-dev \
libgtest-dev \
libopenmpi-dev \
&& \
apt-get clean && \
rm -rf /var/lib/apt/lists/*
Expand Down Expand Up @@ -52,24 +53,6 @@ RUN CMAKE_VERSION=3.22.4 && \
rm cmake*
ENV PATH=${CMAKE_DIR}/bin:$PATH

# Install Open MPI
ENV OPENMPI_DIR=/opt/openmpi
RUN OPENMPI_VERSION=4.0.4 && \
OPENMPI_VERSION_SHORT=4.0 && \
OPENMPI_SHA1=50861c22a4b92ca2e069cd49d756dd96c659bfa8 && \
OPENMPI_URL=https://download.open-mpi.org/release/open-mpi/v${OPENMPI_VERSION_SHORT}/openmpi-${OPENMPI_VERSION}.tar.bz2 && \
OPENMPI_ARCHIVE=openmpi-${OPENMPI_VERSION}.tar.bz2 && \
SCRATCH_DIR=/scratch && mkdir -p ${SCRATCH_DIR} && cd ${SCRATCH_DIR} && \
wget --quiet ${OPENMPI_URL} --output-document=${OPENMPI_ARCHIVE} && \
echo "${OPENMPI_SHA1} ${OPENMPI_ARCHIVE}" | sha1sum -c && \
mkdir -p openmpi && \
tar -xf ${OPENMPI_ARCHIVE} -C openmpi --strip-components=1 && \
mkdir -p build && cd build && \
../openmpi/configure --prefix=${OPENMPI_DIR} CFLAGS=-w && \
make -j${NPROCS} install && \
rm -rf ${SCRATCH_DIR}
ENV PATH=${OPENMPI_DIR}/bin:$PATH

# Install Kokkos
ARG KOKKOS_VERSION=4.1.00
ARG KOKKOS_OPTIONS="-DKokkos_ENABLE_HIP=ON -DKokkos_ENABLE_LIBDL=OFF -DKokkos_ARCH_VEGA908=ON"
Expand Down

0 comments on commit 2342dd2

Please sign in to comment.