Fix Kokkos compile error with NVIDIA Hopper GPU
This commit is contained in:
@ -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; }
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user