Update Tools/machines/perlmutter-nersc/Containerfile

Co-authored-by: Axel Huebl <axel.huebl@plasma.ninja>
ECP-WarpX · Dec 13, 2024 · fc8f29b · fc8f29b
1 parent 28419a0
commit fc8f29b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/Tools/machines/perlmutter-nersc/Containerfile b/Tools/machines/perlmutter-nersc/Containerfile
@@ -6,6 +6,12 @@ ENV DEBIAN_FRONTEND=noninteractive \
     SW_DIR=/opt/software \
     FORCE_UNSAFE_CONFIGURE=1
 
+# Perlmutter A100 compilation optimization
+ENV AMREX_CUDA_ARCH=8.0 \
+    CUDAARCHS=80 \
+    CXXFLAGS="-march=znver3" \
+    CFLAGS="-march=znver3"
+
 # Install essential system dependencies including MPI libraries
 RUN apt-get update && apt-get install -y --no-install-recommends \
     autoconf \