diff --git a/CMakeLists.txt b/CMakeLists.txt index 75bf71c1d..06fc7d3ef 100755 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -21,11 +21,11 @@ if(${CUDA_VERSION_MAJOR} VERSION_GREATER_EQUAL "11") 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 "7") OR (${CUDA_VERSION_MAJOR} VERSION_GREATER_EQUAL "12")) +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") option(ENABLE_FP8 "ENABLE_FP8" OFF) if(ENABLE_FP8) - message("CUDA_VERSION ${CUDA_VERSION_MAJOR}.${CUDA_VERSION_MINOR} is greater or equal than 11.7, enable -DENABLE_FP8 flag") + message("CUDA_VERSION ${CUDA_VERSION_MAJOR}.${CUDA_VERSION_MINOR} is greater or equal than 11.8, enable -DENABLE_FP8 flag") endif() endif()