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