Add sm_120 for blackwell (#5903)

2025-04-30 11:45:24 +08:00
parent 1698e94e67
commit 403b855a22
1 changed files with 1 additions and 0 deletions
--- 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