add FP8 to header test

NVIDIA · Jan 30, 2025 · fa4d829 · fa4d829
1 parent c1b40f8
commit fa4d829
Showing 1 changed file with 42 additions and 30 deletions.
diff --git a/cmake/header_test.cu.in b/cmake/header_test.cu.in
@@ -7,9 +7,9 @@
 // a potential macro collision and halts.
 //
 // Hacky way to build a string, but it works on all tested platforms.
-#define CCCL_HEADER_MACRO_CHECK(MACRO, HEADER)                                      \
-  CCCL_HEADER_MACRO_CHECK_IMPL(Identifier MACRO should not be used from Thrust      \
-                          headers due to conflicts with HEADER macros.)
+#define CCCL_HEADER_MACRO_CHECK(MACRO, HEADER) \
+  CCCL_HEADER_MACRO_CHECK_IMPL(                \
+    Identifier MACRO should not be used from Thrust headers due to conflicts with HEADER macros.)
 
 // Use raw platform macros instead of the CCCL macros since we
 // don't want to #include any headers other than the one being tested.
@@ -18,61 +18,73 @@
 #if defined(_MSC_VER) // MSVC
 
 // Fake up an error for MSVC
-#define CCCL_HEADER_MACRO_CHECK_IMPL(msg)                                           \
-  /* Print message that looks like an error: */                                     \
-  __pragma(message(__FILE__ ":" CCCL_HEADER_MACRO_CHECK_IMPL0(__LINE__)             \
-                   ": error: " #msg))                                               \
-  /* abort compilation due to static_assert or syntax error: */                     \
-  static_assert(false, #msg);
-#define CCCL_HEADER_MACRO_CHECK_IMPL0(x) CCCL_HEADER_MACRO_CHECK_IMPL1(x)
-#define CCCL_HEADER_MACRO_CHECK_IMPL1(x) #x
+#  define CCCL_HEADER_MACRO_CHECK_IMPL(msg)                                                                           \
+    /* Print message that looks like an error: */                                                                     \
+    __pragma(message(__FILE__ ":" CCCL_HEADER_MACRO_CHECK_IMPL0(__LINE__) ": error: " #msg)) /* abort compilation due \
+                                                                                                to static_assert or   \
+                                                                                                syntax error: */      \
+      static_assert(false, #msg);
+#  define CCCL_HEADER_MACRO_CHECK_IMPL0(x) CCCL_HEADER_MACRO_CHECK_IMPL1(x)
+#  define CCCL_HEADER_MACRO_CHECK_IMPL1(x) #x
 
 #elif defined(__clang__) || defined(__GNUC__)
 
 // GCC/clang are easy:
-#define CCCL_HEADER_MACRO_CHECK_IMPL(msg) CCCL_HEADER_MACRO_CHECK_IMPL0(GCC error #msg)
-#define CCCL_HEADER_MACRO_CHECK_IMPL0(expr) _Pragma(#expr)
+#  define CCCL_HEADER_MACRO_CHECK_IMPL(msg)   CCCL_HEADER_MACRO_CHECK_IMPL0(GCC error #msg)
+#  define CCCL_HEADER_MACRO_CHECK_IMPL0(expr) _Pragma(#expr)
 
 #endif // msvc vs. the world
 
 // May be defined to skip macro check for certain configurations.
 #ifndef CCCL_IGNORE_HEADER_MACRO_CHECKS
 
 // complex.h conflicts
-#define I CCCL_HEADER_MACRO_CHECK('I', complex.h)
+#  define I CCCL_HEADER_MACRO_CHECK('I', complex.h)
 
 // windows.h conflicts
-#define small CCCL_HEADER_MACRO_CHECK('small', windows.h)
+#  define small CCCL_HEADER_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(...) CCCL_HEADER_MACRO_CHECK('min', windows.h)
-//#define max(...) CCCL_HEADER_MACRO_CHECK('max', windows.h)
+// #define min(...) CCCL_HEADER_MACRO_CHECK('min', windows.h)
+// #define max(...) CCCL_HEADER_MACRO_CHECK('max', windows.h)
 
-#ifdef _WIN32
+#  ifdef _WIN32
 // On Windows, make sure any include of Windows.h (e.g. via NVTX) does not define the checked macros
-#  define WIN32_LEAN_AND_MEAN
-#endif // _WIN32
+#    define WIN32_LEAN_AND_MEAN
+#  endif // _WIN32
 
 // termios.h conflicts (NVIDIA/thrust#1547)
-#define B0 CCCL_HEADER_MACRO_CHECK("B0", termios.h)
+#  define B0 CCCL_HEADER_MACRO_CHECK("B0", termios.h)
 
 #endif // CCCL_IGNORE_HEADER_MACRO_CHECKS
 
 #include <@header@>
 
+#if defined(CCCL_DISABLE_NVFP8_SUPPORT)
+#  if defined(__CUDA_FP8_TYPES_EXIST__)
+#    error We should not include cuda_fp8.h when FP8 support is disabled
+#  endif // __CUDA_FP16_TYPES_EXIST__
+#endif // CCCL_DISABLE_BF16_SUPPORT
+
 #if defined(CCCL_DISABLE_BF16_SUPPORT)
-#if defined(__CUDA_BF16_TYPES_EXIST__)
-#error We should not include cuda_bf16.h when BF16 support is disabled
-#endif // __CUDA_BF16_TYPES_EXIST__
+#  if defined(__CUDA_BF16_TYPES_EXIST__)
+#    error We should not include cuda_bf16.h when BF16 support is disabled
+#  endif // __CUDA_BF16_TYPES_EXIST__
+#  if defined(__CUDA_FP8_TYPES_EXIST__)
+#    error We should not include cuda_fp8.h when BF16 support is disabled
+#  endif // __CUDA_FP16_TYPES_EXIST__
 #endif // CCCL_DISABLE_BF16_SUPPORT
 
 #if defined(CCCL_DISABLE_FP16_SUPPORT)
-#if defined(__CUDA_FP16_TYPES_EXIST__)
-#error We should not include cuda_fp16.h when half support is disabled
-#endif // __CUDA_FP16_TYPES_EXIST__
-#if defined(__CUDA_BF16_TYPES_EXIST__)
-#error We should not include cuda_bf16.h when half support is disabled
-#endif // __CUDA_BF16_TYPES_EXIST__
+#  if defined(__CUDA_FP8_TYPES_EXIST__)
+#    error We should not include cuda_fp8.h when half support is disabled
+#  endif // __CUDA_FP16_TYPES_EXIST__
+#  if defined(__CUDA_FP16_TYPES_EXIST__)
+#    error We should not include cuda_fp16.h when half support is disabled
+#  endif // __CUDA_FP16_TYPES_EXIST__
+#  if defined(__CUDA_BF16_TYPES_EXIST__)
+#    error We should not include cuda_bf16.h when half support is disabled
+#  endif // __CUDA_BF16_TYPES_EXIST__
 #endif // CCCL_DISABLE_FP16_SUPPORT