From 5cba6674a44dfd8d259693c143267b2fa9398e7b Mon Sep 17 00:00:00 2001
From: Nicolas Richart <nicolas.richart@epfl.ch>
Date: Wed, 31 Jul 2024 13:36:10 +0200
Subject: [PATCH] kokkos-kernels specs to compile
---
stacks/pinot-noir/packages_kuma-h100.yaml | 3 +++
stacks/pinot-noir/packages_kuma-l40s.yaml | 3 +++
stacks/pinot-noir/packages_stack.yaml | 4 ++--
3 files changed, 8 insertions(+), 2 deletions(-)
diff --git a/stacks/pinot-noir/packages_kuma-h100.yaml b/stacks/pinot-noir/packages_kuma-h100.yaml
index d0fa7ff..42304b3 100644
--- a/stacks/pinot-noir/packages_kuma-h100.yaml
+++ b/stacks/pinot-noir/packages_kuma-h100.yaml
@@ -31,6 +31,9 @@ packages:
- spec: '+cuda +cuda_lambda cuda_arch=90 +cuda_uvm +wrapper ~openmptarget'
when: '%gcc'
+ kokkos-kernels:
+ require: ['+cuda', '+cublas', '+cusolver', '+cusparse', 'cuda_arch=90']
+
nvshmem:
require:
- spec: '+cuda cuda_arch=90 +ibgda +nccl +ucx +mpi'
diff --git a/stacks/pinot-noir/packages_kuma-l40s.yaml b/stacks/pinot-noir/packages_kuma-l40s.yaml
index 30371ea..34bc2fd 100644
--- a/stacks/pinot-noir/packages_kuma-l40s.yaml
+++ b/stacks/pinot-noir/packages_kuma-l40s.yaml
@@ -31,6 +31,9 @@ packages:
- spec: '+cuda +cuda_lambda cuda_arch=89 +cuda_uvm +wrapper ~openmptarget'
when: '%gcc'
+ kokkos-kernels:
+ require: ['+cuda', '+cublas', '+cusolver', '+cusparse', 'cuda_arch=89']
+
nvshmem:
require:
- spec: '+cuda cuda_arch=89 +ibgda +nccl +ucx +mpi'
diff --git a/stacks/pinot-noir/packages_stack.yaml b/stacks/pinot-noir/packages_stack.yaml
index 4a5c6bd..a5350e1 100644
--- a/stacks/pinot-noir/packages_stack.yaml
+++ b/stacks/pinot-noir/packages_stack.yaml
@@ -83,8 +83,8 @@ packages:
require: ['+openmp', '+numactl', '+memkind', '~threads',
'+aggressive_vectorization', '@4.3.01']
- kokkos-kernel:
- prefer: ['+openmp', '~threads', '~ipo', '@4.3.01']
+ kokkos-kernels:
+ require: ['@4.3.01', '+openmp', '~threads', '~ipo']
kokkos-nvcc-wrapper:
prefer: ['@4.3.01']
--
GitLab