Fix Kokkos compile error with NVIDIA Hopper GPU

This commit is contained in:
Stan Gerald Moore
2023-02-15 10:02:25 -07:00
parent 4dac68ef72
commit f775de1884

View File

@ -290,7 +290,8 @@ KOKKOS_INLINE_FUNCTION
// Go in this branch if CUDA version is >= 11.0.0 and less than 11.1.0 or if the // Go in this branch if CUDA version is >= 11.0.0 and less than 11.1.0 or if the
// architecture is not Ampere // architecture is not Ampere
#if CUDA_VERSION >= 11000 && \ #if CUDA_VERSION >= 11000 && \
(CUDA_VERSION < 11010 || !defined(KOKKOS_ARCH_AMPERE)) (CUDA_VERSION < 11010 || \
!(defined(KOKKOS_ARCH_AMPERE) || defined(KOKKOS_ARCH_HOPPER)))
KOKKOS_INLINE_FUNCTION KOKKOS_INLINE_FUNCTION
bhalf_t cast_to_bhalf(bhalf_t val) { return val; } bhalf_t cast_to_bhalf(bhalf_t val) { return val; }