533145c458
This patch adds the necessary support to provide `assert` functionality through the GPU `libc` implementation. This implementation creates a special-case GPU implementation rather than relying on the common version. This is because the GPU has special considerings for printing. The assertion is printed out in chunks with `write_to_stderr`, however when combined with the GPU execution model this causes 32+ threads to all execute in-lock step. Meaning that we'll get a horribly fragmented message. Furthermore, potentially thousands of threads could hit the assertion at once and try to print even if we had it all in one `printf`. This is solved by having a one-time lock that each thread group / wave / warp will attempt to claim. We only let one thread group pass through while the others simply stop executing. Finally only the first thread in that group will do the printing until we finally abort execution. Reviewed By: sivachandra Differential Revision: https://reviews.llvm.org/D159296
78 lines
1.4 KiB
TableGen
78 lines
1.4 KiB
TableGen
include "config/public_api.td"
|
|
|
|
include "spec/stdc.td"
|
|
include "spec/posix.td"
|
|
include "spec/gpu_ext.td"
|
|
include "spec/gnu_ext.td"
|
|
include "spec/llvm_libc_ext.td"
|
|
|
|
def AssertMacro : MacroDef<"assert"> {
|
|
let Defn = [{
|
|
#undef assert
|
|
|
|
#ifdef NDEBUG
|
|
#define assert(e) (void)0
|
|
#else
|
|
|
|
#define assert(e) \
|
|
((e) ? (void)0 : __assert_fail(#e, __FILE__, __LINE__, __PRETTY_FUNCTION__))
|
|
#endif
|
|
}];
|
|
}
|
|
|
|
def StaticAssertMacro : MacroDef<"static_assert"> {
|
|
let Defn = [{
|
|
#ifndef __cplusplus
|
|
#undef static_assert
|
|
#define static_assert _Static_assert
|
|
#endif
|
|
}];
|
|
}
|
|
|
|
def AssertAPI : PublicAPI<"assert.h"> {
|
|
let Macros = [
|
|
AssertMacro,
|
|
StaticAssertMacro,
|
|
];
|
|
}
|
|
|
|
def StringAPI : PublicAPI<"string.h"> {
|
|
let Types = ["size_t"];
|
|
}
|
|
|
|
def StdlibAPI : PublicAPI<"stdlib.h"> {
|
|
let Types = [
|
|
"div_t",
|
|
"ldiv_t",
|
|
"lldiv_t",
|
|
"size_t",
|
|
"__atexithandler_t",
|
|
];
|
|
}
|
|
|
|
def FenvAPI: PublicAPI<"fenv.h"> {
|
|
let Types = ["fenv_t"];
|
|
}
|
|
|
|
def StdIOAPI : PublicAPI<"stdio.h"> {
|
|
let Macros = [
|
|
SimpleMacroDef<"_IOFBF", "0">,
|
|
SimpleMacroDef<"_IOLBF", "1">,
|
|
SimpleMacroDef<"_IONBF", "2">,
|
|
SimpleMacroDef<"EOF", "-1">,
|
|
];
|
|
let Types = ["size_t", "FILE"];
|
|
}
|
|
|
|
def IntTypesAPI : PublicAPI<"inttypes.h"> {
|
|
let Types = ["imaxdiv_t"];
|
|
}
|
|
|
|
def TimeAPI : PublicAPI<"time.h"> {
|
|
let Types = [
|
|
"clock_t",
|
|
"time_t",
|
|
"struct timespec",
|
|
];
|
|
}
|