From 851d15fd2ab5401716342ebe653399b15dcc02c5 Mon Sep 17 00:00:00 2001 From: Fabio Luporini Date: Fri, 21 Apr 2023 14:58:00 +0100 Subject: [PATCH] install: Make mpi4py portable across intel and amd --- docker/Dockerfile.nvidia | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docker/Dockerfile.nvidia b/docker/Dockerfile.nvidia index 87c83aee75..a8811d5519 100644 --- a/docker/Dockerfile.nvidia +++ b/docker/Dockerfile.nvidia @@ -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" @@ -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" @@ -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"