diff --git a/docker/Dockerfile.devito b/docker/Dockerfile.devito index aeda36d615..7908497ca5 100644 --- a/docker/Dockerfile.devito +++ b/docker/Dockerfile.devito @@ -26,8 +26,8 @@ RUN python3 -m venv /venv && \ /venv/bin/pip install --no-cache-dir --upgrade pip && \ /venv/bin/pip install --no-cache-dir jupyter && \ /venv/bin/pip install --no-cache-dir wheel && \ - eval "$MPI4PY_FLAGS /venv/bin/pip install --no-cache-dir mpi4py" && \ - /venv/bin/pip install --no-cache-dir -e /app/devito[extras,mpi,tests] && \ + eval "$MPI4PY_FLAGS /venv/bin/pip install --no-cache-dir -r /app/devito/requirements-mpi.txt" && \ + /venv/bin/pip install --no-cache-dir -e /app/devito[extras,tests] && \ rm -rf ~/.cache/pip # Usefull utilities diff --git a/docker/Dockerfile.intel b/docker/Dockerfile.intel index 48757d9776..24e57ea147 100644 --- a/docker/Dockerfile.intel +++ b/docker/Dockerfile.intel @@ -59,6 +59,8 @@ RUN apt-get update -y && apt-get dist-upgrade -y && \ # Development packages libigc-dev intel-igc-cm libigdfcl-dev libigfxcmrt-dev level-zero-dev +ENV MPI4PY_FLAGS='. /opt/intel/oneapi/setvars.sh intel64 && ' + ############################################################## # ICC image # This is a legacy setup that is not built anymore but kept for reference @@ -70,12 +72,13 @@ RUN apt-get update -y && apt-get install -y intel-oneapi-compiler-dpcpp-cpp-and- rm -rf /var/lib/apt/lists/* # Devito config +ENV I_MPI_CC="icc" +ENV I_MPI_CXX="icpc" ENV DEVITO_ARCH="icc" ENV DEVITO_LANGUAGE="openmp" ENV DEVITO_PLATFORM="intel64" # MPICC compiler for mpi4py ENV MPICC=mpiicc -ENV MPI4PY_FLAGS='. /opt/intel/oneapi/setvars.sh && CFLAGS="-cc=icc"' ############################################################## # ICX OpenMP image @@ -87,11 +90,12 @@ RUN apt-get update -y && apt-get install -y intel-oneapi-compiler-dpcpp-cpp inte rm -rf /var/lib/apt/lists/* # Devito config +ENV I_MPI_CC="icx" +ENV I_MPI_CXX="icpx" ENV DEVITO_ARCH="icx" ENV DEVITO_LANGUAGE="openmp" # MPICC compiler for mpi4py ENV MPICC=mpiicc -ENV MPI4PY_FLAGS='. /opt/intel/oneapi/setvars.sh && CFLAGS="-cc=icx"' ############################################################## # ICX SYCL CPU image @@ -102,6 +106,7 @@ FROM icx as cpu-sycl ENV DEVITO_ARCH="sycl" ENV DEVITO_LANGUAGE="sycl" ENV DEVITO_PLATFORM="intel64" +ENV MPICC=sycl ############################################################## # ICX SYCL GPU image @@ -112,3 +117,4 @@ FROM icx as gpu-sycl ENV DEVITO_ARCH="sycl" ENV DEVITO_LANGUAGE="sycl" ENV DEVITO_PLATFORM="intelgpuX" +ENV MPICC=sycl