Add sm_120 for blackwell (#5903)

403b855a · zhjunqin · GitHub · 1698e94e · 403b855a
Unverified Commit 403b855a authored Apr 30, 2025 by zhjunqin Committed by GitHub Apr 29, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

sgl-kernel/CMakeLists.txt sgl-kernel/CMakeLists.txt +1 -0

No files found.
--- 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