From 403b855a22fe2787737cc0a7ad6df8dc467ef772 Mon Sep 17 00:00:00 2001 From: zhjunqin Date: Wed, 30 Apr 2025 11:45:24 +0800 Subject: [PATCH] Add sm_120 for blackwell (#5903) --- sgl-kernel/CMakeLists.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/sgl-kernel/CMakeLists.txt b/sgl-kernel/CMakeLists.txt index 375b35d1c..6fefea9d4 100755 --- a/sgl-kernel/CMakeLists.txt +++ b/sgl-kernel/CMakeLists.txt @@ -132,6 +132,7 @@ 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" ) else() list(APPEND SGL_KERNEL_CUDA_FLAGS