Merge pull request #2007 from lucbv/bhalf_t_fix

bhalf_t fix for isnan function
kokkos · Oct 19, 2023 · b9f3d78 · b9f3d78
2 parents ab0a32d + 72a8d71
commit b9f3d78
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/common/src/Kokkos_ArithTraits.hpp b/common/src/Kokkos_ArithTraits.hpp
@@ -1336,9 +1336,13 @@ class ArithTraits<Kokkos::Experimental::bhalf_t> {
   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