Add cuda to pm-gpu spack yaml files

E3SM-Project · Aug 5, 2024 · 3c312b0 · 3c312b0
1 parent 1245ee3
commit 3c312b0
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/mache/spack/pm-gpu_gnugpu_mpich.yaml b/mache/spack/pm-gpu_gnugpu_mpich.yaml
@@ -110,6 +110,18 @@ spack:
         - craype/2.7.19
         - libfabric/1.15.2.0
       buildable: false
+    cuda:
+      externals:
+      - spec: [email protected]
+        prefix: /opt/nvidia/hpc_sdk/Linux_x86_64/22.7/cuda/11.7
+        modules:
+        - PrgEnv-gnu/8.3.3
+        - gcc/11.2.0
+        - cudatoolkit/11.7
+        - craype-accel-nvidia80
+        - craype/2.7.19
+        - libfabric/1.15.2.0
+      buildable: false
     cray-mpich:
       externals:
       - spec: [email protected]

diff --git a/mache/spack/pm-gpu_nvidiagpu_mpich.yaml b/mache/spack/pm-gpu_nvidiagpu_mpich.yaml
@@ -98,6 +98,19 @@ spack:
       - spec: [email protected]
         prefix: /usr
       buildable: false
+    cuda:
+      externals:
+      - spec: [email protected]
+        prefix: /opt/nvidia/hpc_sdk/Linux_x86_64/22.7/cuda/11.7
+        modules:
+        - PrgEnv-nvidia
+        - nvidia/22.7
+        - cudatoolkit/11.7
+        - craype-accel-nvidia80
+        - gcc-mixed/11.2.0
+        - craype-x86-milan
+        - libfabric
+      buildable: false
     cray-mpich:
       externals:
       - spec: [email protected]