EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

introduce CUB in sgl-kernel (#2887)

Browse Source
This commit is contained in:
Xiaoyu Zhang
2025-01-14 19:48:59 +08:00
committed by GitHub
parent f5c6c66794
commit f005758f2b
3 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
.gitmodules vendored
View File

@@ -1,3 +1,6 @@
[submodule "sgl-kernel/3rdparty/cutlass"] [submodule "sgl-kernel/3rdparty/cutlass"]
path = sgl-kernel/3rdparty/cutlass path = sgl-kernel/3rdparty/cutlass
url = https://github.com/NVIDIA/cutlass.git url = https://github.com/NVIDIA/cutlass.git
[submodule "sgl-kernel/3rdparty/cub"]
path = sgl-kernel/3rdparty/cub
url = https://github.com/NVIDIA/cub.git

1
sgl-kernel/3rdparty/cub vendored Submodule

Submodule sgl-kernel/3rdparty/cub added at 0fc3c37016

2
sgl-kernel/CMakeLists.txt
View File

@@ -9,6 +9,7 @@ set(CMAKE_CUDA_STANDARD 17)
set(CMAKE_CUDA_STANDARD_REQUIRED ON) set(CMAKE_CUDA_STANDARD_REQUIRED ON)
set(CUTLASS_DIR "3rdparty/cutlass") set(CUTLASS_DIR "3rdparty/cutlass")
set(CUB_DIR "3rdparty/cub")
# Set CUDA architectures # Set CUDA architectures
set(CMAKE_CUDA_ARCHITECTURES "75;80;86;89;90") set(CMAKE_CUDA_ARCHITECTURES "75;80;86;89;90")
@@ -43,6 +44,7 @@ target_include_directories(_kernels
${TORCH_INCLUDE_DIRS} ${TORCH_INCLUDE_DIRS}
${CUTLASS_DIR}/include ${CUTLASS_DIR}/include
${CUTLASS_DIR}/tools/util/include ${CUTLASS_DIR}/tools/util/include
${CUB_DIR}/cub
) )
target_link_libraries(_kernels target_link_libraries(_kernels