diff --git a/sgl-kernel/CMakeLists.txt b/sgl-kernel/CMakeLists.txt index f8481491a..d0c7bbee3 100644 --- a/sgl-kernel/CMakeLists.txt +++ b/sgl-kernel/CMakeLists.txt @@ -161,18 +161,14 @@ if (ENABLE_BELOW_SM90) ) endif() -if ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "13.0" OR SGL_KERNEL_ENABLE_SM100A) - list(APPEND SGL_KERNEL_CUDA_FLAGS - "-gencode=arch=compute_100,code=sm_110" - "-gencode=arch=compute_100a,code=sm_110a" - ) -elseif ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "12.8" OR SGL_KERNEL_ENABLE_SM100A) +if ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "12.8" OR SGL_KERNEL_ENABLE_SM100A) list(APPEND SGL_KERNEL_CUDA_FLAGS "-gencode=arch=compute_100,code=sm_100" - "-gencode=arch=compute_100,code=sm_101" - "-gencode=arch=compute_100,code=sm_101a" "-gencode=arch=compute_100a,code=sm_100a" + "-gencode=arch=compute_101,code=sm_101" + "-gencode=arch=compute_101a,code=sm_101a" "-gencode=arch=compute_120,code=sm_120" + "-gencode=arch=compute_120a,code=sm_120a" ) else() list(APPEND SGL_KERNEL_CUDA_FLAGS