diff --git a/stacks/pinot-noir/packages_kuma-h100.yaml b/stacks/pinot-noir/packages_kuma-h100.yaml index d0fa7ff31faf29e65bd7c6336f1593cbe74f88a7..42304b3dcaf547340b4426251ad277dab2f8b2f7 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 30371eafc2dec2a1b4559334fff2ff39a003578d..34bc2fd955271b5a15ba80c97658fb4a39197580 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 4a5c6bdccd6c11cd98f1704ec52884a261830118..a5350e144ad4076af834ccd8a9a8b164bfb5627e 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']