From 72a8d71fd2c8e6dd5014f6f039d0481f6250bf8e Mon Sep 17 00:00:00 2001 From: Luc Berger-Vergiat Date: Wed, 18 Oct 2023 14:43:27 -0600 Subject: [PATCH] bhalf_t fix for isnan function --- common/src/Kokkos_ArithTraits.hpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/common/src/Kokkos_ArithTraits.hpp b/common/src/Kokkos_ArithTraits.hpp index 17296185e7..75c0951e10 100644 --- a/common/src/Kokkos_ArithTraits.hpp +++ b/common/src/Kokkos_ArithTraits.hpp @@ -1336,9 +1336,13 @@ class ArithTraits { static KOKKOS_FUNCTION mag_type rmax() { return Kokkos::Experimental::cast_to_bhalf(KOKKOSKERNELS_IMPL_BF16_MAX); } +#else +#if defined(KOKKOS_ENABLE_SYCL) || defined(KOKKOS_ENABLE_HIP) + KOKKOSKERNELS_ARITHTRAITS_HALF_FP(KOKKOS_FUNCTION) #else KOKKOSKERNELS_ARITHTRAITS_REAL_FP(KOKKOS_FUNCTION) #endif +#endif }; #endif // #if defined(KOKKOS_BHALF_T_IS_FLOAT) && !KOKKOS_BHALF_T_IS_FLOAT