[jit_kernel] Temporarily Skip Flaky JIT Kernel GDN Test and Add PR Label (#20436)
This commit is contained in:
@@ -11,6 +11,11 @@ sgl-kernel:
|
|||||||
- changed-files:
|
- changed-files:
|
||||||
- any-glob-to-any-file: 'sgl-kernel/**/*'
|
- 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
|
||||||
documentation:
|
documentation:
|
||||||
- changed-files:
|
- 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 CUTEDSL_AVAILABLE, reason="CuTe DSL not available")
|
||||||
@pytest.mark.skipif(not TRITON_AVAILABLE, reason="Triton kernel 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])
|
@pytest.mark.parametrize("B", [16, 128])
|
||||||
def test_cutedsl_gdn_precision(B: int):
|
def test_cutedsl_gdn_precision(B: int):
|
||||||
"""Test precision of CuTe DSL GDN kernel against Triton reference."""
|
"""Test precision of CuTe DSL GDN kernel against Triton reference."""
|
||||||
|
|||||||
Reference in New Issue
Block a user