[Diffusion] Refactor diffusion JIT kernel test layout and narrow CI triggers (#21385)

This commit is contained in:
Xiaoyu Zhang
2026-03-26 15:02:02 +08:00
committed by GitHub
parent 79db3bec34
commit 7ca015fe65
11 changed files with 22 additions and 12 deletions
+6 -2
View File
@@ -192,8 +192,12 @@ def run_a_suite(args):
# JIT kernel tests and benchmarks (live alongside kernel source)
jit_kernel_dir = os.path.join(repo_root, "python", "sglang", "jit_kernel")
files += glob.glob(os.path.join(jit_kernel_dir, "tests", "test_*.py"))
files += glob.glob(os.path.join(jit_kernel_dir, "benchmark", "bench_*.py"))
files += glob.glob(
os.path.join(jit_kernel_dir, "tests", "**", "test_*.py"), recursive=True
)
files += glob.glob(
os.path.join(jit_kernel_dir, "benchmark", "**", "bench_*.py"), recursive=True
)
# Strict: all discovered files must have proper registration
sanity_check = True