ursa: add arborx for UFS WM env, add oneAPI compiler modules for MPI and MKL - #2118
Open
rickgrubin-noaa wants to merge 3 commits into
Open
ursa: add arborx for UFS WM env, add oneAPI compiler modules for MPI and MKL#2118rickgrubin-noaa wants to merge 3 commits into
rickgrubin-noaa wants to merge 3 commits into
Conversation
9 tasks
climbfuji
reviewed
Sep 4, 2026
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
This PR adds:
arborxto the UFS WM env package recipeufs_weather_model_env/package.pyDescription
Package
arborxis now added as a variant forufs_weather_model_env, following on with the addition ofkokkosandkokkos-kernelsin PR #2088.Per discussion with Ben Koziol re:
CICE/CECE/CATChemcomponents, envs were built for oneAPI and GCC witharborx, and with and withoutCUDAsupport. Testing leads to a preference for addingarborxbut excludingCUDAsupport at this time.Like
kokkos/kokkos-kernels,arborxis added as a variant toufs_weather_model_env/package.py(default=False) and is enabled only forufs_weather_model_env.arborx/kokkos/kokkos-kernelsare spec'd to build with%cxx=gccbecause for H100 GPUs,kokkosrelies onnvcc_wrapperfor CMakeCUDAsupport, and GCC is the primary target host compiler suggested by NVIDIA, as it is known to provide broad stability. The test envs bear this out.As well, for the backends chosen to be built for
arborxandkokkos, GCC is preferred.Comments are provided in
configs/sites/tier1/ursa/packages.yamlto turn onCUDAsupport if desired in the future.Lastly, for oneAPI package yaml files, an entry for the compiler is added to both the MPI and MKL specs
modulessection, so as to have MPI and MKL module successfully located and loaded at build time.Dependencies
None
Issues addressed
Related to #2066 and likely closes it. See also #2110 and PR #2119
Applications affected
UFS WM
Systems affected
ursa only
Testing
Checklist