Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "cac7adab11a6fcbd592f5b8c7fa48ce194758bde"
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
Hide 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)
)
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
......
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