Update CMakeLists.txt

d676654e · zhuwenwen · cdd4dc9f · d676654e
Commit d676654e authored May 24, 2024 by zhuwenwen
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

CMakeLists.txt CMakeLists.txt +4 -4

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -16,10 +16,10 @@ project(FasterTransformer LANGUAGES CXX CUDA)

 find_package(CUDA 10.2 REQUIRED)

-if(${CUDA_VERSION_MAJOR} VERSION_GREATER_EQUAL "11")
-  add_definitions("-DENABLE_BF16")
-  message("CUDA_VERSION ${CUDA_VERSION_MAJOR}.${CUDA_VERSION_MINOR} is greater or equal than 11.0, enable -DENABLE_BF16 flag")
-endif()
+#if(${CUDA_VERSION_MAJOR} VERSION_GREATER_EQUAL "11")
+  #add_definitions("-DENABLE_BF16")
+  #message("CUDA_VERSION ${CUDA_VERSION_MAJOR}.${CUDA_VERSION_MINOR} is greater or equal than 11.0, enable -DENABLE_BF16 flag")
+#endif()

 if((${CUDA_VERSION_MAJOR} VERSION_GREATER_EQUAL "11" AND ${CUDA_VERSION_MINOR} VERSION_GREATER_EQUAL "8") OR (${CUDA_VERSION_MAJOR} VERSION_GREATER_EQUAL "12"))
  add_definitions("-DENABLE_FP8")