"vscode:/vscode.git/clone" did not exist on "5a2451047fd8156a86bcfa6fd157a3f31328726e"
Unverified Commit fc8c8e50 authored by Qiaolin Yu's avatar Qiaolin Yu Committed by GitHub
Browse files

Integrate triton_kernels in sgl-kernel (#8762)

parent 9bd4872a
...@@ -66,6 +66,16 @@ FetchContent_Declare( ...@@ -66,6 +66,16 @@ FetchContent_Declare(
GIT_SHALLOW OFF GIT_SHALLOW OFF
) )
FetchContent_Populate(repo-deepgemm) FetchContent_Populate(repo-deepgemm)
# Triton
FetchContent_Declare(
repo-triton
GIT_REPOSITORY "https://github.com/triton-lang/triton"
GIT_TAG 8f9f695ea8fde23a0c7c88e4ab256634ca27789f
GIT_SHALLOW OFF
)
FetchContent_Populate(repo-triton)
# flashinfer # flashinfer
FetchContent_Declare( FetchContent_Declare(
repo-flashinfer repo-flashinfer
...@@ -413,3 +423,9 @@ install(DIRECTORY "${repo-cutlass_SOURCE_DIR}/include/cute/" ...@@ -413,3 +423,9 @@ install(DIRECTORY "${repo-cutlass_SOURCE_DIR}/include/cute/"
install(DIRECTORY "${repo-cutlass_SOURCE_DIR}/include/cutlass/" install(DIRECTORY "${repo-cutlass_SOURCE_DIR}/include/cutlass/"
DESTINATION "deep_gemm/include/cutlass") DESTINATION "deep_gemm/include/cutlass")
# triton_kernels
install(DIRECTORY "${repo-triton_SOURCE_DIR}/python/triton_kernels/triton_kernels/"
DESTINATION "triton_kernels"
PATTERN ".git*" EXCLUDE
PATTERN "__pycache__" EXCLUDE)
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