Skip to content
Snippets Groups Projects
Commit 14800658 authored by Nicolas Richart's avatar Nicolas Richart
Browse files

Manage to get 1 cuda 1 ucx 1 openmpi

parent 79cfb7f0
No related branches found
No related tags found
1 merge request!3Feat/stack pinot noir
...@@ -75,7 +75,6 @@ ...@@ -75,7 +75,6 @@
"ca-certificates-mozilla", "ca-certificates-mozilla",
"check", "check",
"cmake", "cmake",
"cuda",
"curl", "curl",
"diffutils", "diffutils",
"expat", "expat",
......
definitions: definitions:
- cuda_system_codes: - cuda_system_codes: []
- cuda
- cudnn
- cuda_serial_codes: - cuda_serial_codes:
- cuda
- cudnn
- nccl - nccl
- cuda_mpi_codes:
- nvshmem - nvshmem
packages: packages:
all: all:
compiler:
- gcc
- aocc
- nvhpc
providers: providers:
mpi: [openmpi, intel-oneapi-mpi] mpi: [openmpi, intel-oneapi-mpi]
# iconv: [libiconv]
jpeg: [libjpeg] jpeg: [libjpeg]
zlib-api: [zlib-ng] zlib-api: [zlib-ng]
# require:
# - spec: "^libiconv" glibc:
# when: "^iconv" prefer: ['%gcc@11.2.1']
zlib-ng: zlib-ng:
require: ['build_system=autotools'] require: ['build_system=autotools']
packages: packages:
all: all:
target: ["zen4"] target: ["zen4"]
prefer: ['target=zen4']
require: require:
- "~rocm" - "~rocm"
- spec: cuda_arch=90 target=zen4 - spec: cuda_arch=90
when: '+cuda' when: '+cuda'
- spec: '+cuda' - spec: '+cuda'
when: '%gcc' when: '%gcc'
# --------------------------------------------------------------------------- # ---------------------------------------------------------------------------
cuda:
prefer:
- spec: "^gcc-runtime@12.3.0"
when: "%gcc"
- spec: "%gcc@12.3.0 target=zen4"
when: "%gcc"
hypre: hypre:
require: require:
- spec: '+cuda cuda_arch=90 +unified-memory' - spec: '+cuda cuda_arch=90 +unified-memory'
...@@ -27,7 +35,7 @@ packages: ...@@ -27,7 +35,7 @@ packages:
openmpi: openmpi:
require: require:
- spec: '+cuda cuda_arch=90 target=zen4' - spec: '+cuda cuda_arch=90'
petsc: petsc:
require: require:
...@@ -52,7 +60,7 @@ packages: ...@@ -52,7 +60,7 @@ packages:
ucx: ucx:
require: require:
- spec: '+xpmem +cma +rdmacm +rc +ud +dc +verbs +ib_hw_tm +gdrcopy +cuda cuda_arch=90 ~rocm target=zen4' - spec: '+xpmem +cma +rdmacm +rc +ud +dc +verbs +ib_hw_tm +gdrcopy +cuda cuda_arch=90 ~rocm'
# --------------------------------------------------------------------------- # ---------------------------------------------------------------------------
# Externals # Externals
......
packages: packages:
all: all:
compiler:
- gcc
- aocc
- nvhpc
providers:
mpi: [openmpi, intel-oneapi-mpi]
iconv: [libiconv]
jpeg: [libjpeg]
zlib-api: [zlib-ng]
require: require:
- spec: '+ipo' - spec: '+ipo'
when: '%gcc' when: '%gcc'
...@@ -19,6 +10,9 @@ packages: ...@@ -19,6 +10,9 @@ packages:
adios2: adios2:
require: ['+kokkos'] require: ['+kokkos']
arpack-ng:
prefer: ['+mpi']
boost: boost:
prefer: ['~mpi'] prefer: ['~mpi']
require: ['cxxstd=14', '+icu', '+python', '+numpy', '+atomic', '+chrono', require: ['cxxstd=14', '+icu', '+python', '+numpy', '+atomic', '+chrono',
...@@ -61,11 +55,11 @@ packages: ...@@ -61,11 +55,11 @@ packages:
prefer: ['~cluster'] prefer: ['~cluster']
kokkos: kokkos:
prefer: ['+openmp', '+numactl', '+memkind', '+threads', prefer: ['+openmp', '+numactl', '+memkind', '~threads',
'+aggressive_vectorization'] '+aggressive_vectorization']
kokkos-kernel: kokkos-kernel:
prefer: ['+openmp', '+threads'] prefer: ['+openmp', '~threads']
lammps: lammps:
require: ['build_type=Release', '+asphere', '+atc', '+body', '+class2', require: ['build_type=Release', '+asphere', '+atc', '+body', '+class2',
...@@ -123,6 +117,10 @@ packages: ...@@ -123,6 +117,10 @@ packages:
scotch: scotch:
prefer: ['~mpi'] prefer: ['~mpi']
slepc:
require:
- spec: '^arpack-ng +mpi'
suite-sparse: suite-sparse:
require: require:
- spec: '+graphblas' - spec: '+graphblas'
......
...@@ -80,10 +80,8 @@ spack: ...@@ -80,10 +80,8 @@ spack:
# Serial codes # Serial codes
# ------------------------------------------------------------------------- # -------------------------------------------------------------------------
- serial_codes: - serial_codes:
- bwa
- fftw +openmp ~mpi - fftw +openmp ~mpi
- fftw ~openmp ~mpi - fftw ~openmp ~mpi
- eigen
- gsl - gsl
- hdf5@1.14.1 ~mpi - hdf5@1.14.1 ~mpi
- hisat2 - hisat2
...@@ -107,8 +105,10 @@ spack: ...@@ -107,8 +105,10 @@ spack:
- voropp - voropp
- gcc_serial_codes: - gcc_serial_codes:
- bwa
- cistem ^fftw~mpi - cistem ^fftw~mpi
- ctffind ^fftw ~mpi - ctffind ^fftw ~mpi
- eigen
- ffmpeg +libx264 - ffmpeg +libx264
- glpk+gmp - glpk+gmp
- hwloc - hwloc
...@@ -129,6 +129,8 @@ spack: ...@@ -129,6 +129,8 @@ spack:
- cuda_serial_codes: [] - cuda_serial_codes: []
- cuda_mpi_codes: []
# ------------------------------------------------------------------------- # -------------------------------------------------------------------------
# Blas dependent codes # Blas dependent codes
# ------------------------------------------------------------------------- # -------------------------------------------------------------------------
...@@ -256,6 +258,7 @@ spack: ...@@ -256,6 +258,7 @@ spack:
#- quantum-espresso +mpi +scalapack +gipaw hdf5=parallel #- quantum-espresso +mpi +scalapack +gipaw hdf5=parallel
- gcc_mpi_blas_codes: - gcc_mpi_blas_codes:
- armadillo +hdf5 ^arpack-ng +mpi ^hdf5 +mpi
- gmsh +mpi +eigen +openmp +hdf5 ~fltk ~opencascade ~med ^mmg ~vtk - gmsh +mpi +eigen +openmp +hdf5 ~fltk ~opencascade ~med ^mmg ~vtk
- cpmd ~openmp - cpmd ~openmp
- elmerfem +mumps +openmp +hypre - elmerfem +mumps +openmp +hypre
...@@ -282,7 +285,7 @@ spack: ...@@ -282,7 +285,7 @@ spack:
- lammps - lammps
- opencv - opencv
- openfoam +metis - openfoam +metis
- paraview +shared +python +hdf5 ~osmesa +opengl2 +egl - paraview +shared +python +hdf5 +opengl2
- py-fenics-dolfinx - py-fenics-dolfinx
- py-h5py +mpi - py-h5py +mpi
- py-horovod - py-horovod
...@@ -397,6 +400,11 @@ spack: ...@@ -397,6 +400,11 @@ spack:
- [$^mpi_gcc] - [$^mpi_gcc]
- [$%compiler_gcc] - [$%compiler_gcc]
- matrix:
- [$cuda_mpi_codes]
- [$^mpi_gcc]
- [$%compiler_gcc]
- matrix: - matrix:
- [$mpi_blas_codes] - [$mpi_blas_codes]
- [$^blas_gcc] - [$^blas_gcc]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment