[jit_kernel] Temporarily Skip Flaky JIT Kernel GDN Test and Add PR Label (#20436)

This commit is contained in:
Xiaoyu Zhang
2026-03-13 09:34:22 +08:00
committed by GitHub
parent 65dd08153d
commit 7ecf07b8f4
2 changed files with 11 additions and 0 deletions

5
.github/labeler.yml vendored
View File

@@ -11,6 +11,11 @@ sgl-kernel:
- changed-files:
- any-glob-to-any-file: 'sgl-kernel/**/*'
# JIT kernel specific
jit-kernel:
- changed-files:
- any-glob-to-any-file: 'python/sglang/jit_kernel/**/*'
# Documentation
documentation:
- changed-files:

View File

@@ -47,6 +47,12 @@ def run_triton_kernel(A_log, dt_bias, q, k, v, a, b, initial_state, indices, sca
@pytest.mark.skipif(not CUTEDSL_AVAILABLE, reason="CuTe DSL not available")
@pytest.mark.skipif(not TRITON_AVAILABLE, reason="Triton kernel not available")
@pytest.mark.skip(
reason=(
"Temporary CI workaround: CuTe DSL GDN precision is currently unstable "
"against the Triton reference and needs follow-up investigation."
)
)
@pytest.mark.parametrize("B", [16, 128])
def test_cutedsl_gdn_precision(B: int):
"""Test precision of CuTe DSL GDN kernel against Triton reference."""