Add sm_120 for blackwell (#5903)

zhjunqin · web-flow · commit 403b855a22fe · 2025-04-29T20:45:24.000-07:00
diff --git 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

Original file line number	Diff line number	Diff line change
`@@ -132,6 +132,7 @@ if ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "12.8" OR SGL_KERNEL_ENABLE_SM100A)`
`132`	`132`	`list(APPEND SGL_KERNEL_CUDA_FLAGS`
`133`	`133`	`"-gencode=arch=compute_100,code=sm_100"`
`134`	`134`	`"-gencode=arch=compute_100a,code=sm_100a"`
	`135`	`+ "-gencode=arch=compute_120,code=sm_120"`
`135`	`136`	`)`
`136`	`137`	`else()`
`137`	`138`	`list(APPEND SGL_KERNEL_CUDA_FLAGS`