diff --git a/sgl-kernel/CMakeLists.txt b/sgl-kernel/CMakeLists.txt index 095ad47f7..87389ec4b 100644 --- a/sgl-kernel/CMakeLists.txt +++ b/sgl-kernel/CMakeLists.txt @@ -194,26 +194,20 @@ endif() 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_100a,code=sm_100a" - "-gencode=arch=compute_120,code=sm_120" "-gencode=arch=compute_120a,code=sm_120a" ) # refer sm_121, sm_110 and sm_101 description https://github.com/pytorch/pytorch/pull/156176 if ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "13.0") list(APPEND SGL_KERNEL_CUDA_FLAGS - "-gencode=arch=compute_103,code=sm_103" "-gencode=arch=compute_103a,code=sm_103a" - "-gencode=arch=compute_110,code=sm_110" "-gencode=arch=compute_110a,code=sm_110a" - "-gencode=arch=compute_121,code=sm_121" "-gencode=arch=compute_121a,code=sm_121a" "--compress-mode=size" ) else() list(APPEND SGL_KERNEL_CUDA_FLAGS - "-gencode=arch=compute_101,code=sm_101" "-gencode=arch=compute_101a,code=sm_101a" ) endif()