Skip to content

Commit

Permalink
Merge pull request #2115 from devitocodes/heal-mpi4py
Browse files Browse the repository at this point in the history 
install: Make mpi4py portable across intel and amd
  • Loading branch information
@mloubout
mloubout authored Apr 21, 2023
2 parents 0c9987a + 851d15f commit f5505fa
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions docker/Dockerfile.nvidia
View file
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,7 @@ FROM sdk-base as nvc
ADD docker/nvdashboard.json /app/nvdashboard.json

# mpi4py
RUN CFLAGS=-noswitcherror /venv/bin/pip install --no-cache-dir mpi4py && rm -rf ~/.cache/pip
RUN CC=nvc CFLAGS="-noswitcherror -tp=px" /venv/bin/pip install --no-cache-dir mpi4py && rm -rf ~/.cache/pip

ENV DEVITO_ARCH="nvc"
ENV DEVITO_PLATFORM="nvidiaX"
Expand All @@ -145,7 +145,7 @@ FROM sdk-base as nvcc
ADD docker/nvdashboard.json /app/nvdashboard.json

# mpi4py
RUN CFLAGS=-noswitcherror /venv/bin/pip install --no-cache-dir mpi4py && rm -rf ~/.cache/pip
RUN CC=nvc CFLAGS="-noswitcherror -tp=px" /venv/bin/pip install --no-cache-dir mpi4py && rm -rf ~/.cache/pip

ENV DEVITO_ARCH="cuda"
ENV DEVITO_PLATFORM="nvidiaX"
Expand All @@ -157,7 +157,7 @@ ENV DEVITO_LANGUAGE="cuda"
FROM sdk-base as nvc-host

# mpi4py
RUN CFLAGS=-noswitcherror /venv/bin/pip install --no-cache-dir mpi4py && rm -rf ~/.cache/pip
RUN CC=nvc CFLAGS="-noswitcherror -tp=px" /venv/bin/pip install --no-cache-dir mpi4py && rm -rf ~/.cache/pip

ENV DEVITO_ARCH="nvc"
ENV DEVITO_PLATFORM="cpu64"
Expand Down

0 comments on commit f5505fa

Please sign in to comment.