File size: 2,374 Bytes
8ae5fc5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | // This source file checks that:
// 1) Header <cub/thread/thread_load.cuh> compiles without error.
// 2) Common macro collisions with platform/system headers are avoided.
// Define CUB_MACRO_CHECK(macro, header), which emits a diagnostic indicating
// a potential macro collision and halts.
//
// Use raw platform checks instead of the CUB_HOST_COMPILER macros since we
// don't want to #include any headers other than the one being tested.
//
// This is only implemented for MSVC/GCC/Clang.
#if defined(_MSC_VER) // MSVC
// Fake up an error for MSVC
#define CUB_MACRO_CHECK_IMPL(msg) \
/* Print message that looks like an error: */ \
__pragma(message(__FILE__ ":" CUB_MACRO_CHECK_IMPL0(__LINE__) \
": error: " #msg)) \
/* abort compilation due to static_assert or syntax error: */ \
static_assert(false, #msg);
#define CUB_MACRO_CHECK_IMPL0(x) CUB_MACRO_CHECK_IMPL1(x)
#define CUB_MACRO_CHECK_IMPL1(x) #x
#elif defined(__clang__) || defined(__GNUC__)
// GCC/clang are easy:
#define CUB_MACRO_CHECK_IMPL(msg) CUB_MACRO_CHECK_IMPL0(GCC error #msg)
#define CUB_MACRO_CHECK_IMPL0(expr) _Pragma(#expr)
#endif
// Hacky way to build a string, but it works on all tested platforms.
#define CUB_MACRO_CHECK(MACRO, HEADER) \
CUB_MACRO_CHECK_IMPL(Identifier MACRO should not be used from CUB \
headers due to conflicts with HEADER macros.)
// complex.h conflicts
#define I CUB_MACRO_CHECK('I', complex.h)
// windows.h conflicts
#define small CUB_MACRO_CHECK('small', windows.h)
// We can't enable these checks without breaking some builds -- some standard
// library implementations unconditionally `#undef` these macros, which then
// causes random failures later.
// Leaving these commented out as a warning: Here be dragons.
//#define min(...) CUB_MACRO_CHECK('min', windows.h)
//#define max(...) CUB_MACRO_CHECK('max', windows.h)
// termios.h conflicts (NVIDIA/thrust#1547)
#define B0 CUB_MACRO_CHECK("B0", termios.h)
#include <cub/thread/thread_load.cuh>
#if defined(CUB_DISABLE_BF16_SUPPORT)
#if defined(__CUDA_BF16_TYPES_EXIST__)
#error CUB should not include cuda_bf16.h when BF16 support is disabled
#endif
#endif
|