introduce CUB in sgl-kernel (#2887)

f005758f · Xiaoyu Zhang · GitHub · f5c6c667 · f005758f · 0fc3c370
Unverified Commit f005758f authored Jan 14, 2025 by Xiaoyu Zhang Committed by GitHub Jan 14, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

.gitmodules .gitmodules +3 -0

sgl-kernel/3rdparty/cub sgl-kernel/3rdparty/cub +1 -0

sgl-kernel/CMakeLists.txt sgl-kernel/CMakeLists.txt +2 -0

No files found.
--- a/.gitmodules
+++ b/.gitmodules
 [submodule "sgl-kernel/3rdparty/cutlass"]
 	path = sgl-kernel/3rdparty/cutlass
 	url = https://github.com/NVIDIA/cutlass.git
+[submodule "sgl-kernel/3rdparty/cub"]
+	path = sgl-kernel/3rdparty/cub
+	url = https://github.com/NVIDIA/cub.git
--- a/cub @ 0fc3c370
+++ b/cub @ 0fc3c370
+Subproject commit 0fc3c3701632a4be906765b73be20a9ad0da603d
--- a/sgl-kernel/CMakeLists.txt
+++ b/sgl-kernel/CMakeLists.txt
@@ -9,6 +9,7 @@ set(CMAKE_CUDA_STANDARD 17)
 set(CMAKE_CUDA_STANDARD_REQUIRED ON)

 set(CUTLASS_DIR "3rdparty/cutlass")
+set(CUB_DIR "3rdparty/cub")

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

 target_link_libraries(_kernels