From c1dd9a95990b505ace3ca11dcd69e676c7673895 Mon Sep 17 00:00:00 2001
From: DarkSharpness <76582120+DarkSharpness@users.noreply.github.com>
Date: Tue, 25 Nov 2025 15:19:17 +0800
Subject: [PATCH] [Fix] JIT kernel dependencies in other platforms (#13889)

---
 python/pyproject_cpu.toml   | 3 +++
 python/pyproject_other.toml | 3 +++
 python/pyproject_xpu.toml   | 3 +++
 3 files changed, 9 insertions(+)

diff --git a/python/pyproject_cpu.toml b/python/pyproject_cpu.toml
index 50406c14b..257ac35ed 100644
--- a/python/pyproject_cpu.toml
+++ b/python/pyproject_cpu.toml
@@ -100,6 +100,9 @@ dev = ["sglang[test]"]
   "srt/mem_cache/storage/hf3fs/hf3fs_utils.cpp",
   "srt/speculative/cpp_ngram/*.cpp",
   "srt/speculative/cpp_ngram/*.h",
+  "jit_kernel/include/sgl_kernel/*.h",
+  "jit_kernel/include/sgl_kernel/*.cuh",
+  "jit_kernel/csrc/*.cuh"
 ]
 
 [tool.setuptools.packages.find]
diff --git a/python/pyproject_other.toml b/python/pyproject_other.toml
index cdc0fe1c2..55007a25e 100755
--- a/python/pyproject_other.toml
+++ b/python/pyproject_other.toml
@@ -122,6 +122,9 @@ dev_hpu = ["sglang[all_hpu]", "sglang[test]"]
   "srt/mem_cache/storage/hf3fs/hf3fs_utils.cpp",
   "srt/speculative/cpp_ngram/*.cpp",
   "srt/speculative/cpp_ngram/*.h",
+  "jit_kernel/include/sgl_kernel/*.h",
+  "jit_kernel/include/sgl_kernel/*.cuh",
+  "jit_kernel/csrc/*.cuh"
 ]
 
 [tool.setuptools.packages.find]
diff --git a/python/pyproject_xpu.toml b/python/pyproject_xpu.toml
index 465f2b0f6..3e88356dc 100644
--- a/python/pyproject_xpu.toml
+++ b/python/pyproject_xpu.toml
@@ -104,6 +104,9 @@ dev = ["sglang[test]"]
   "srt/mem_cache/storage/hf3fs/hf3fs_utils.cpp",
   "srt/speculative/cpp_ngram/*.cpp",
   "srt/speculative/cpp_ngram/*.h",
+  "jit_kernel/include/sgl_kernel/*.h",
+  "jit_kernel/include/sgl_kernel/*.cuh",
+  "jit_kernel/csrc/*.cuh"
 ]
 
 [tool.setuptools.packages.find]