@@ -8,27 +8,27 @@ ARG PLATFORM=cupy
88ARG CUDAVERSION=12.4
99
1010# Pull from mambaforge and install XML and ssh
11- FROM condaforge/mambaforge as base
11+ FROM condaforge/mambaforge AS base
1212ENV DEBIAN_FRONTEND=noninteractive
1313RUN apt-get update && apt-get install -y libxml2 ssh
1414
1515# Pull from base image and install OpenMPI/MPICH
16- FROM base as mpi
16+ FROM base AS mpi
1717ARG MPI
1818RUN mamba install -n base -c conda-forge ${MPI}
1919
2020# Pull from MPI build install core dependencies
21- FROM base as core
21+ FROM base AS core
2222COPY ./dependencies_core.yml ./dependencies.yml
2323RUN mamba env update -n base -f dependencies.yml
2424
2525# Pull from MPI build and install full dependencies
26- FROM mpi as full
26+ FROM mpi AS full
2727COPY ./dependencies_full.yml ./dependencies.yml
2828RUN mamba env update -n base -f dependencies.yml
2929
3030# Pull from MPI build and install accelerate/pycuda dependencies
31- FROM mpi as pycuda
31+ FROM mpi AS pycuda
3232ARG CUDAVERSION
3333COPY ./ptypy/accelerate/cuda_pycuda/dependencies.yml ./dependencies.yml
3434COPY ./cufft/dependencies.yml ./dependencies_cufft.yml
@@ -37,7 +37,7 @@ RUN mamba install cuda-version=${CUDAVERSION} && \
3737 mamba env update -n base -f dependencies_cufft.yml
3838
3939# Pull from MPI build and install accelerate/cupy dependencies
40- FROM mpi as cupy
40+ FROM mpi AS cupy
4141ARG CUDAVERSION
4242COPY ./ptypy/accelerate/cuda_cupy/dependencies.yml ./dependencies.yml
4343COPY ./cufft/dependencies.yml ./dependencies_cufft.yml
@@ -46,7 +46,7 @@ RUN mamba install cuda-version=${CUDAVERSION} && \
4646 mamba env update -n base -f dependencies_cufft.yml
4747
4848# Pull from platform specific image and install ptypy
49- FROM ${PLATFORM} as build
49+ FROM ${PLATFORM} AS build
5050COPY pyproject.toml ./
5151COPY ./templates ./templates
5252COPY ./benchmark ./benchmark
@@ -55,19 +55,19 @@ COPY ./ptypy ./ptypy
5555RUN pip install .
5656
5757# For core/full build, no post processing needed
58- FROM build as core-post
59- FROM build as full-post
58+ FROM build AS core-post
59+ FROM build AS full-post
6060
6161# For pycuda build, install filtered cufft
62- FROM build as pycuda-post
62+ FROM build AS pycuda-post
6363RUN pip install ./cufft
6464
6565# For pycuda build, install filtered cufft
66- FROM build as cupy-post
66+ FROM build AS cupy-post
6767RUN pip install ./cufft
6868
6969# Platform specific runtime container
70- FROM ${PLATFORM}-post as runtime
70+ FROM ${PLATFORM}-post AS runtime
7171
7272# Run PtyPy run script as entrypoint
7373ENTRYPOINT ["ptypy.cli" ]
0 commit comments