From a63c0cafd35a79cba1aa00de3ad6b724a0bd1b1e Mon Sep 17 00:00:00 2001
From: Nicolas Richart <nicolas.richart@epfl.ch>
Date: Fri, 28 Jun 2024 20:55:01 +0200
Subject: [PATCH] Moving all cuda reference in cuda systems
---
stacks/pinot-noir/packages_jed.yaml | 10 +++++++---
1 file changed, 7 insertions(+), 3 deletions(-)
diff --git a/stacks/pinot-noir/packages_jed.yaml b/stacks/pinot-noir/packages_jed.yaml
index dcc04e3..c260f3e 100644
--- a/stacks/pinot-noir/packages_jed.yaml
+++ b/stacks/pinot-noir/packages_jed.yaml
@@ -15,12 +15,12 @@ packages:
# ---------------------------------------------------------------------------
hypre:
require:
- - spec: '+cuda cuda_arch=90'
+ - spec: '+cuda cuda_arch=90 +unified-memory'
when: '%gcc'
kokkos:
require:
- - spec: '+cuda cuda_arch=90 +cuda_uvm'
+ - spec: '+cuda cuda_arch=90 +cuda_uvm +wrapper'
when: '%gcc'
openmpi:
@@ -32,6 +32,10 @@ packages:
- spec: '+cuda cuda_arch=90'
when: '%gcc'
+ py-tensorflow:
+ require:
+ - spec: '+cuda cuda_arch=90 +nccl'
+
py-torch:
require:
- spec: '+cuda cuda_arch=90 +nccl +cudnn'
@@ -46,7 +50,7 @@ packages:
ucx:
require:
- - spec: '+xpmem +cma +rdmacm +rc +ud +dc +verbs +ib_hw_tm +vfs +gdrcopy +cuda cuda_arch=90 target=icelake ~rocm'
+ - spec: '+xpmem +cma +rdmacm +rc +ud +dc +verbs +ib_hw_tm +gdrcopy +cuda cuda_arch=90 target=icelake ~rocm'
# ---------------------------------------------------------------------------
# Externals
--
GitLab