zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

kokkos: disable CUDA_MALLOC_ASYNC on cray-mpich (#42661)

Browse Source 
Co-authored-by: Daniel Arndt <arndtd@ornl.gov>
This commit is contained in:
Richard Berger 2024-02-19 05:52:48 -07:00 committed by GitHub
parent a1f90620c3
commit c2b116175b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
var/spack/repos/builtin/packages/kokkos/package.py
View File

@ -356,6 +356,17 @@ def cmake_args(self):
if self.spec.satisfies("%oneapi") or self.spec.satisfies("%intel"):
options.append(self.define("CMAKE_CXX_FLAGS", "-fp-model=precise"))
# Kokkos 4.2.00+ changed the default to Kokkos_ENABLE_IMPL_CUDA_MALLOC_ASYNC=on
# which breaks GPU-aware with Cray-MPICH
# See https://github.com/kokkos/kokkos/pull/6402
# TODO: disable this once Cray-MPICH is fixed
if (
self.spec.satisfies("@4.2.00:")
and "mpi" in self.spec
and self.spec["mpi"].name == "cray-mpich"
):
options.append(self.define("Kokkos_ENABLE_IMPL_CUDA_MALLOC_ASYNC", False))
# Remove duplicate options
return lang.dedupe(options)