Add bf8 cmake bindings

CMakeLists.txt

@@ -15,6 +15,12 @@ if (DTYPES)
     if (DTYPES MATCHES "fp8")
         add_definitions(-DCK_ENABLE_FP8)
         set(CK_ENABLE_FP8 "ON")
+        add_compile_options(-Wno-bit-int-extension)
+    endif()
+    if (DTYPES MATCHES "bf8")
+        add_definitions(-DCK_ENABLE_BF8)
+        set(CK_ENABLE_BF8 "ON")
+        add_compile_options(-Wno-bit-int-extension)
     endif()
     if (DTYPES MATCHES "fp16")
         add_definitions(-DCK_ENABLE_FP16)
@@ -34,8 +40,9 @@ if (DTYPES)
     endif()
     message("DTYPES macro set to ${DTYPES}")
 else()
-    add_definitions(-DCK_ENABLE_INT8 -DCK_ENABLE_FP8 -DCK_ENABLE_FP16 -DCK_ENABLE_FP32 -DCK_ENABLE_FP64 -DCK_ENABLE_BF16)
+    add_definitions(-DCK_ENABLE_INT8 -DCK_ENABLE_FP8 -DCK_ENABLE_BF8 -DCK_ENABLE_FP16 -DCK_ENABLE_FP32 -DCK_ENABLE_FP64 -DCK_ENABLE_BF16)
     set(CK_ENABLE_ALL_DTYPES "ON")
+    add_compile_options(-Wno-bit-int-extension) # enable fp8 and bf8
 endif()
 
 if(DL_KERNELS)
@@ -365,6 +372,10 @@ IF(IS_DIRECTORY "${PROJECT_SOURCE_DIR}/library/src/tensor_operation_instance/gpu
         #message("fp8 instance found!")
         set(add_inst 1)
     endif()
+    if("${cmake_instance}" MATCHES "DTYPES MATCHES \"bf8\" " AND DTYPES MATCHES "bf8")
+        #message("bf8 instance found!")
+        set(add_inst 1)
+    endif()
     if("${cmake_instance}" MATCHES "DTYPES MATCHES \"fp16\"" AND DTYPES MATCHES "fp16")
         #message("fp16 instance found!")
         set(add_inst 1)

example/01_gemm/CMakeLists.txt

@@ -70,5 +70,7 @@ if(DTYPES MATCHES "fp8" OR NOT DEFINED DTYPES)
   endif()
 endif()
 
-add_example_executable(example_gemm_xdl_fp16_f8 gemm_xdl_fp16_f8.cpp)
-add_dependencies(example_gemm_xdl example_gemm_xdl_fp16_f8)
+if(DTYPES MATCHES "fp8" OR DTYPES MATCHES "fp16" OR NOT DEFINED DTYPES)
+  add_example_executable(example_gemm_xdl_fp16_f8 gemm_xdl_fp16_f8.cpp)
+  add_dependencies(example_gemm_xdl example_gemm_xdl_fp16_f8)
+endif()

include/ck/config.h.in

@@ -43,6 +43,9 @@
 #ifndef CK_ENABLE_FP8
 #define CK_ENABLE_FP8 "ON"
 #endif
+#ifndef CK_ENABLE_BF8
+#define CK_ENABLE_BF8 "ON"
+#endif
 #ifndef CK_ENABLE_FP16
 #define CK_ENABLE_FP16 "ON"
 #endif
@@ -66,6 +69,10 @@
 #cmakedefine CK_ENABLE_FP8 @CK_ENABLE_FP8@
 #endif
 
+#ifndef CK_ENABLE_BF8
+#cmakedefine CK_ENABLE_BF8 @CK_ENABLE_BF8@
+#endif
+
 #ifndef CK_ENABLE_FP16
 #cmakedefine CK_ENABLE_FP16 @CK_ENABLE_FP16@
 #endif

test/data_type/CMakeLists.txt

@@ -3,8 +3,12 @@ if (USE_BITINT_EXTENSION_INT4)
   target_link_libraries(test_int4 PRIVATE utility)
 endif()
 
-add_gtest_executable(test_f8 f8.cpp)
-target_link_libraries(test_f8 PRIVATE utility)
+if(DTYPES MATCHES "fp8" OR NOT DEFINED DTYPES)
+  add_gtest_executable(test_f8 f8.cpp)
+  target_link_libraries(test_f8 PRIVATE utility)
+endif()
 
-add_gtest_executable(test_bf8 bf8.cpp)
-target_link_libraries(test_bf8 PRIVATE utility)
+if(DTYPES MATCHES "bf8" OR NOT DEFINED DTYPES)
+  add_gtest_executable(test_bf8 bf8.cpp)
+  target_link_libraries(test_bf8 PRIVATE utility)
+endif()