Skip to content

ursa: add arborx for UFS WM env, add oneAPI compiler modules for MPI and MKL #2118

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
rickgrubin-noaa wants to merge 3 commits into
base: develop
Choose a base branch
from
Open

ursa: add arborx for UFS WM env, add oneAPI compiler modules for MPI and MKL #2118

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
780ebf5
Add arborx for UFS WM env with kokkos, add oneAPI compiler modules fo…
rickgrubin-noaa Sep 3, 2026
2309e6b
Remove unnecessary when= argument for kokkos / arborx requirements
rickgrubin-noaa Sep 4, 2026
4a0ea28
Merge branch 'develop' into feature/kokkos-arborx
rickgrubin-noaa Sep 4, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions configs/common/packages.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,12 @@ packages:
require:
- zlib
# Individual package settings
arborx:
require:
- '@2.1'
- +mpi
- +openmp
- +serial
awscli-v2:
require:
- ~examples
Expand Down
12 changes: 9 additions & 3 deletions configs/sites/tier1/ursa/packages.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,17 +9,23 @@ packages:
- '@1.2.13'

# Modifications of common packages
# GPUs on ursa are H100
ufs-weather-model-env:
require::
- +kokkos
- +arborx

# GPUs on ursa are H100
# add: '+cuda cuda_arch=90' to kokkos / kokkos-kernels / arborx if building with CUDA support
# add: '+wrapper' to kokkos
kokkos:
require:
- '+cuda cuda_arch=90'
- '%cxx=gcc'
kokkos-kernels:
require:
- '+cuda cuda_arch=90'
- '%cxx=gcc'
arborx:
require:
- '%cxx=gcc'

# All other packages listed alphabetically
autoconf:
Expand Down
2 changes: 2 additions & 0 deletions configs/sites/tier1/ursa/packages_oneapi-2025.3.1-hpcx.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@ packages:
- spec: hpcx-mpi@2.18.1
prefix: /apps/hpcx-v2.18.1-gcc-mlnx_ofed-redhat9-cuda12-x86_64/ompi-mt-oneapi2025
modules:
- intel-oneapi-compilers/2025.3.1
- hpc-x/2.18.1-mt-oneapi2025
intel-oneapi-compilers:
buildable: False
Expand All @@ -37,6 +38,7 @@ packages:
- spec: intel-oneapi-mkl@2025.3
prefix: /apps/spack-2024-12/linux-rocky9-x86_64/oneapi-2025.3.1/intel-oneapi-mkl-2025.3.0-z5owhveid7qlq4erbay2seqzs65y3etu
modules:
- intel-oneapi-compilers/2025.3.1
- intel-oneapi-mkl/2025.3.0
gcc:
buildable: False
Expand Down
2 changes: 2 additions & 0 deletions configs/sites/tier1/ursa/packages_oneapi-2025.3.1.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -30,13 +30,15 @@ packages:
- spec: intel-oneapi-mpi@2021.17
prefix: /apps/spack-2024-12/linux-rocky9-x86_64/oneapi-2025.3.1/intel-oneapi-mpi-2021.17.1-emmz2zv6lx5ypzmlc3csouqffu7mwnwr
modules:
- intel-oneapi-compilers/2025.3.1
- intel-oneapi-mpi/2021.17.1
intel-oneapi-mkl:
buildable: False
externals:
- spec: intel-oneapi-mkl@2025.3
prefix: /apps/spack-2024-12/linux-rocky9-x86_64/oneapi-2025.3.1/intel-oneapi-mkl-2025.3.0-z5owhveid7qlq4erbay2seqzs65y3etu
modules:
- intel-oneapi-compilers/2025.3.1
- intel-oneapi-mkl/2025.3.0
intel-oneapi-tbb:
buildable: False
Expand Down
7 changes: 6 additions & 1 deletion repos/spack_stack/spack_repo/spack_stack/packages/ufs_weather_model_env/package.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,8 @@ class UfsWeatherModelEnv(BundlePackage):
)
variant("python", default=True, description="Include extra Python packages")
variant("ncutils", default=True, description="Include extra NetCDF utilities (cprnc and nccmp)")
variant("kokkos", default=False, description="Include kokkos and kokkos-kernels")
variant("kokkos", default=False, description="Enable kokkos and kokkos-kernels")
variant("arborx", default=False, description="Enable arborx")

depends_on("cmake", type="run")
depends_on("python", type="run")
Expand Down Expand Up @@ -63,4 +64,8 @@ class UfsWeatherModelEnv(BundlePackage):
depends_on("kokkos", type="run")
depends_on("kokkos-kernels", type="run")

with when("+arborx"):
depends_on("arborx", type="run")
requires("+kokkos", msg="Arborx requires +kokkos")

# There is no need for install() since there is no code.
Loading