From fc8c8e504156ac8bca3caa4987024b2313cddd87 Mon Sep 17 00:00:00 2001 From: Qiaolin Yu Date: Mon, 4 Aug 2025 12:12:14 -0700 Subject: [PATCH] Integrate triton_kernels in sgl-kernel (#8762) --- sgl-kernel/CMakeLists.txt | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/sgl-kernel/CMakeLists.txt b/sgl-kernel/CMakeLists.txt index b359c76c1..72225e55b 100644 --- a/sgl-kernel/CMakeLists.txt +++ b/sgl-kernel/CMakeLists.txt @@ -66,6 +66,16 @@ FetchContent_Declare( GIT_SHALLOW OFF ) 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 FetchContent_Declare( repo-flashinfer @@ -413,3 +423,9 @@ install(DIRECTORY "${repo-cutlass_SOURCE_DIR}/include/cute/" install(DIRECTORY "${repo-cutlass_SOURCE_DIR}/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)