kokkos: disable CUDA_MALLOC_ASYNC on cray-mpich (#42661)
Co-authored-by: Daniel Arndt <arndtd@ornl.gov>
This commit is contained in:
parent
a1f90620c3
commit
c2b116175b
@ -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)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user