From fc8f29be6964f5d72b7293adf1c4d66140858537 Mon Sep 17 00:00:00 2001
From: U Sinha <46874465+sinha-r@users.noreply.github.com>
Date: Fri, 13 Dec 2024 14:38:21 -0800
Subject: [PATCH] Update Tools/machines/perlmutter-nersc/Containerfile

Co-authored-by: Axel Huebl <axel.huebl@plasma.ninja>
---
 Tools/machines/perlmutter-nersc/Containerfile | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/Tools/machines/perlmutter-nersc/Containerfile b/Tools/machines/perlmutter-nersc/Containerfile
index 9d355bfdc4e..4ad4ad46196 100644
--- 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 \