From 16d76b9f2336a372d51e69566ddd3920a4eac435 Mon Sep 17 00:00:00 2001 From: Ruihang Lai Date: Wed, 25 Jun 2025 22:00:46 -0400 Subject: [PATCH] [CMake] Fix sgl-kernel CMakeLists for Blackwell (#7543) --- sgl-kernel/CMakeLists.txt | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) 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