From ef7c29acd7ef3ec500f23c15e4f5dac88c22951a Mon Sep 17 00:00:00 2001 From: MarcoDWei Date: Thu, 18 Dec 2025 13:26:42 +0800 Subject: [PATCH] Fix issue: ENABLE_BELOW_SM90 cannot be enabled on aarch64 CPU (#12967) --- sgl-kernel/CMakeLists.txt | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/sgl-kernel/CMakeLists.txt b/sgl-kernel/CMakeLists.txt index 9a438591a..5fe330e9a 100644 --- a/sgl-kernel/CMakeLists.txt +++ b/sgl-kernel/CMakeLists.txt @@ -124,14 +124,16 @@ if(CCACHE_FOUND AND ENABLE_CCACHE AND DEFINED ENV{CCACHE_DIR}) set_property(GLOBAL PROPERTY RULE_LAUNCH_LINK "ccache") endif() -# Enable gencode below SM90 -option(ENABLE_BELOW_SM90 "Enable below SM90" ON) - -if (CMAKE_SYSTEM_PROCESSOR MATCHES "aarch64") - set(ENABLE_BELOW_SM90 OFF) +# Configure gencode below SM90 +if(CMAKE_SYSTEM_PROCESSOR MATCHES "aarch64") + set(DEFAULT_ENABLE_BELOW_SM90 OFF) message(STATUS "For aarch64, disable gencode below SM90 by default") +else() + set(DEFAULT_ENABLE_BELOW_SM90 ON) endif() +option(ENABLE_BELOW_SM90 "Enable gencode below SM90" ${DEFAULT_ENABLE_BELOW_SM90}) + include_directories( ${PROJECT_SOURCE_DIR}/include ${PROJECT_SOURCE_DIR}/csrc