Skip to content

GitLab

Unverified Commit 16d76b9f authored by Ruihang Lai's avatar Ruihang Lai Committed by GitHub
Browse files

[CMake] Fix sgl-kernel CMakeLists for Blackwell (#7543)

parent 5c214257
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 8 deletions
sgl-kernel/CMakeLists.txt
View file @ 16d76b9f
...@@ -161,18 +161,14 @@ if (ENABLE_BELOW_SM90) ...@@ -161,18 +161,14 @@ if (ENABLE_BELOW_SM90)
) )
endif() endif()
if ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "13.0" 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_110"
"-gencode=arch=compute_100a,code=sm_110a"
)
elseif ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "12.8" OR SGL_KERNEL_ENABLE_SM100A)
list(APPEND SGL_KERNEL_CUDA_FLAGS list(APPEND SGL_KERNEL_CUDA_FLAGS
"-gencode=arch=compute_100,code=sm_100" "-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_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_120,code=sm_120"
"-gencode=arch=compute_120a,code=sm_120a"
) )
else() else()
list(APPEND SGL_KERNEL_CUDA_FLAGS list(APPEND SGL_KERNEL_CUDA_FLAGS
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment