Skip to content

Commit 9f21e75

Browse files
authored
add Thor & Spark (#5915)
1 parent 7bcd8b1 commit 9f21e75

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

sgl-kernel/CMakeLists.txt

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,9 @@ find_package(CUDAToolkit REQUIRED)
2222
set_property(GLOBAL PROPERTY CUDA_SEPARABLE_COMPILATION ON)
2323

2424
message(STATUS "Detected CUDA_VERSION=${CUDA_VERSION}")
25-
if ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "12.8")
25+
if ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "13.0")
26+
message("CUDA_VERSION ${CUDA_VERSION} >= 13.0")
27+
elseif ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "12.8")
2628
message("CUDA_VERSION ${CUDA_VERSION} >= 12.8")
2729
elseif ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "12.4")
2830
message("CUDA_VERSION ${CUDA_VERSION} >= 12.4")
@@ -128,9 +130,16 @@ option(SGL_KERNEL_ENABLE_FP8 "Enable FP8" ON)
128130
option(SGL_KERNEL_ENABLE_FP4 "Enable FP4" OFF)
129131
option(SGL_KERNEL_ENABLE_FA3 "Enable FA3" OFF)
130132

131-
if ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "12.8" OR SGL_KERNEL_ENABLE_SM100A)
133+
if ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "13.0" OR SGL_KERNEL_ENABLE_SM100A)
134+
list(APPEND SGL_KERNEL_CUDA_FLAGS
135+
"-gencode=arch=compute_100,code=sm_110"
136+
"-gencode=arch=compute_100a,code=sm_110a"
137+
)
138+
elseif ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "12.8" OR SGL_KERNEL_ENABLE_SM100A)
132139
list(APPEND SGL_KERNEL_CUDA_FLAGS
133140
"-gencode=arch=compute_100,code=sm_100"
141+
"-gencode=arch=compute_100,code=sm_101"
142+
"-gencode=arch=compute_100,code=sm_101a"
134143
"-gencode=arch=compute_100a,code=sm_100a"
135144
"-gencode=arch=compute_120,code=sm_120"
136145
)

0 commit comments

Comments
 (0)