Do not rely on builtins that are not host device for complex

NVIDIA · Jun 26, 2023 · fbbaf6b · fbbaf6b
1 parent bf1faa7
commit fbbaf6b
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/libcudacxx/include/cuda/std/detail/libcxx/include/cmath b/libcudacxx/include/cuda/std/detail/libcxx/include/cmath
@@ -613,7 +613,9 @@ _LIBCUDACXX_INLINE_VISIBILITY
 _LIBCUDACXX_CONSTEXPR __enable_if_t<is_floating_point<_A1>::value, bool>
 __constexpr_isnan(_A1 __lcpp_x) _NOEXCEPT
 {
-#if __has_builtin(__builtin_isnan)
+#if defined(_LIBCUDACXX_COMPILER_NVCC) && _LIBCUDACXX_CUDACC_VER < 1108000
+    return __isnan(__lcpp_x);
+#elif __has_builtin(__builtin_isnan)
     return __builtin_isnan(__lcpp_x);
 #else
     return isnan(__lcpp_x);
@@ -633,7 +635,9 @@ _LIBCUDACXX_INLINE_VISIBILITY
 _LIBCUDACXX_CONSTEXPR __enable_if_t<is_floating_point<_A1>::value, bool>
 __constexpr_isinf(_A1 __lcpp_x) _NOEXCEPT
 {
-#if __has_builtin(__builtin_isinf)
+#if defined(_LIBCUDACXX_COMPILER_NVCC) && _LIBCUDACXX_CUDACC_VER < 1108000
+    return __isinf(__lcpp_x);
+#elif __has_builtin(__builtin_isinf)
     return __builtin_isinf(__lcpp_x);
 #else
     return isinf(__lcpp_x);
@@ -653,7 +657,9 @@ _LIBCUDACXX_INLINE_VISIBILITY
 _LIBCUDACXX_CONSTEXPR __enable_if_t<is_floating_point<_A1>::value, bool>
 __constexpr_isfinite(_A1 __lcpp_x) _NOEXCEPT
 {
-#if __has_builtin(__builtin_isfinite)
+#if defined(_LIBCUDACXX_COMPILER_NVCC) && _LIBCUDACXX_CUDACC_VER < 1108000
+    return !__isinf(__lcpp_x) && !__isnan(__lcpp_x);
+#elif __has_builtin(__builtin_isfinite)
     return __builtin_isfinite(__lcpp_x);
 #else
     return isfinite(__lcpp_x);