From 7ecf07b8f40091aa98f1e81f64b6e84bf8bb4a05 Mon Sep 17 00:00:00 2001 From: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com> Date: Fri, 13 Mar 2026 09:34:22 +0800 Subject: [PATCH] [jit_kernel] Temporarily Skip Flaky JIT Kernel GDN Test and Add PR Label (#20436) --- .github/labeler.yml | 5 +++++ python/sglang/jit_kernel/tests/test_cutedsl_gdn.py | 6 ++++++ 2 files changed, 11 insertions(+) diff --git a/.github/labeler.yml b/.github/labeler.yml index 21970fd74..572839b18 100644 --- a/.github/labeler.yml +++ b/.github/labeler.yml @@ -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: diff --git a/python/sglang/jit_kernel/tests/test_cutedsl_gdn.py b/python/sglang/jit_kernel/tests/test_cutedsl_gdn.py index 4139f24ae..2b06ede6f 100644 --- a/python/sglang/jit_kernel/tests/test_cutedsl_gdn.py +++ b/python/sglang/jit_kernel/tests/test_cutedsl_gdn.py @@ -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."""