Skip to content

GitLab

Unverified Commit f005758f authored by Xiaoyu Zhang's avatar Xiaoyu Zhang Committed by GitHub
Browse files

introduce CUB in sgl-kernel (#2887)

parent f5c6c667
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
.gitmodules
View file @ f005758f
[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
cub @ 0fc3c370
Subproject commit 0fc3c3701632a4be906765b73be20a9ad0da603d
sgl-kernel/CMakeLists.txt
View file @ f005758f
...@@ -9,6 +9,7 @@ set(CMAKE_CUDA_STANDARD 17) ...@@ -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 ...@@ -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
......
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