From d21dc02bc5f70bc7dc0d18ba5801ae263565e68e Mon Sep 17 00:00:00 2001 From: byshiue Date: Mon, 23 Jan 2023 21:00:38 +0800 Subject: [PATCH] Update CMakeLists.txt --- CMakeLists.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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()