Update libcudacxx/include/cuda/__type_traits/is_floating_point.h

Co-authored-by: Michael Schellenberger Costa <[email protected]>
NVIDIA · Jan 21, 2025 · f73b1fc · f73b1fc
1 parent 4180ede
commit f73b1fc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/libcudacxx/include/cuda/__type_traits/is_floating_point.h b/libcudacxx/include/cuda/__type_traits/is_floating_point.h
@@ -35,7 +35,8 @@ struct _CCCL_TYPE_VISIBILITY_DEFAULT is_floating_point
 
 #if !defined(_CCCL_NO_VARIABLE_TEMPLATES)
 template <class _Tp>
-_CCCL_INLINE_VAR constexpr bool is_floating_point_v = is_floating_point<_Tp>::value;
+_CCCL_INLINE_VAR constexpr bool is_floating_point_v = _CUDA_VSTD::is_floating_point_v<_CUDA_VSTD::remove_cv_t<_Tp>>
+                                || _CUDA_VSTD::__is_extended_floating_point_v<_CUDA_VSTD::remove_cv_t<_Tp>>;
 #endif // !_CCCL_NO_VARIABLE_TEMPLATES
 
 _LIBCUDACXX_END_NAMESPACE_CUDA