Skip to content

Commit b433add

Browse files
committed
[CUDA][HIP] Add a __device__ version of std::__glibcxx_assert_fail()
libstdc++ 15 uses the non-constexpr function std::__glibcxx_assert_fail() to trigger compilation errors when the __glibcxx_assert(cond) macro is used in a constantly evaluated context. Compilation fails when using code from the libstdc++ (such as std::array) on device code, since these assertions invoke a non-constexpr host function from device code. This patch proposes a cuda wrapper header "bits/c++config.h" which adds a __device__ version of std::__glibcxx_assert_fail().
1 parent 500cccc commit b433add

File tree

2 files changed

+38
-0
lines changed

2 files changed

+38
-0
lines changed

clang/lib/Headers/CMakeLists.txt

+1
Original file line numberDiff line numberDiff line change
@@ -333,6 +333,7 @@ set(cuda_wrapper_files
333333
)
334334

335335
set(cuda_wrapper_bits_files
336+
cuda_wrappers/bits/c++config.h
336337
cuda_wrappers/bits/shared_ptr_base.h
337338
cuda_wrappers/bits/basic_string.h
338339
cuda_wrappers/bits/basic_string.tcc
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,37 @@
1+
// libstdc++ uses the non-constexpr function std::__glibcxx_assert_fail()
2+
// to trigger compilation errors when the __glibcxx_assert(cond) macro
3+
// is used in a constexpr context.
4+
// Compilation fails when using code from the libstdc++ (such as std::array) on
5+
// device code, since these assertions invoke a non-constexpr host function from
6+
// device code.
7+
//
8+
// To work around this issue, we declare our own device version of the function
9+
10+
#ifndef __CLANG_CUDA_WRAPPERS_BITS_CPP_CONFIG
11+
#define __CLANG_CUDA_WRAPPERS_BITS_CPP_CONFIG
12+
13+
#include_next <bits/c++config.h>
14+
15+
#ifdef _LIBCPP_BEGIN_NAMESPACE_STD
16+
_LIBCPP_BEGIN_NAMESPACE_STD
17+
#else
18+
namespace std {
19+
#ifdef _GLIBCXX_BEGIN_NAMESPACE_VERSION
20+
_GLIBCXX_BEGIN_NAMESPACE_VERSION
21+
#endif
22+
#endif
23+
__attribute__((device))
24+
__attribute__((__always_inline__, __visibility__("default"))) inline void
25+
__glibcxx_assert_fail(...) {
26+
__builtin_abort();
27+
}
28+
#ifdef _LIBCPP_END_NAMESPACE_STD
29+
_LIBCPP_END_NAMESPACE_STD
30+
#else
31+
#ifdef _GLIBCXX_BEGIN_NAMESPACE_VERSION
32+
_GLIBCXX_END_NAMESPACE_VERSION
33+
#endif
34+
} // namespace std
35+
#endif
36+
37+
#endif

0 commit comments

Comments
 (0)