From be996b70ac2757c4fdfb68f9d55dd030cbf8db81 Mon Sep 17 00:00:00 2001
From: Nicolas Richart <nicolas.richart@epfl.ch>
Date: Tue, 30 Jul 2024 20:36:14 +0200
Subject: [PATCH] Synchronizing l40s
---
stacks/pinot-noir/packages_kuma-l40s.yaml | 24 ++++++++++++++++++++---
1 file changed, 21 insertions(+), 3 deletions(-)
diff --git a/stacks/pinot-noir/packages_kuma-l40s.yaml b/stacks/pinot-noir/packages_kuma-l40s.yaml
index a1b6aef..5979953 100644
--- a/stacks/pinot-noir/packages_kuma-l40s.yaml
+++ b/stacks/pinot-noir/packages_kuma-l40s.yaml
@@ -1,16 +1,26 @@
packages:
all:
target: ["zen4"]
+ prefer: ["zen4"]
require:
- "~rocm"
- - spec: cuda_arch=89 target=zen4
+ - spec: cuda_arch=89
when: '+cuda'
- spec: '+cuda'
when: '%gcc'
# ---------------------------------------------------------------------------
+ cuda:
+ prefer:
+ - spec: "^gcc-runtime@12.3.0"
+ when: "%gcc"
+ - spec: "%gcc@12.3.0 target=zen4"
+ when: "%gcc"
+ require:
+ - '@12.4.0'
+
hypre:
require:
- spec: '+cuda cuda_arch=89 +unified-memory'
@@ -27,7 +37,10 @@ packages:
openmpi:
require:
- - spec: '+cuda cuda_arch=89 target=zen4'
+ - spec: '+cuda cuda_arch=89'
+
+ papi:
+ require: ['+nvml', '+infiniband', '+cuda']
petsc:
require:
@@ -42,6 +55,11 @@ packages:
require:
- spec: '+cuda cuda_arch=89 +nccl +cudnn'
+ python:
+ require:
+ - spec: 'target=zen4'
+ when: '%gcc@12.3.0'
+
quantum-espresso:
require:
- spec: '~cuda'
@@ -52,7 +70,7 @@ packages:
ucx:
require:
- - spec: '+xpmem +cma +rdmacm +rc +ud +dc +verbs +ib_hw_tm +gdrcopy +cuda cuda_arch=89 ~rocm target=zen4'
+ - spec: '+xpmem +cma +rdmacm +rc +ud +dc +verbs +ib_hw_tm +gdrcopy +cuda cuda_arch=89 ~rocm'
# ---------------------------------------------------------------------------
# Externals
--
GitLab