[jit_kernel] Temporarily Skip Flaky JIT Kernel GDN Test and Add PR Label (#20436)
This commit is contained in:
5
.github/labeler.yml
vendored
5
.github/labeler.yml
vendored
@@ -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:
|
||||
|
||||
@@ -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."""
|
||||
|
||||
Reference in New Issue
Block a user