minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Skip causal_conv1d test with padded batches due to Triton kernel bug (#16715)

Browse Source
This commit is contained in:
Alison Shao
2026-01-08 15:13:13 -08:00
committed by GitHub
parent ee71e773e1
commit a979927727
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/layers/mamba/test_causal_conv1d.py
+6
View File
@@ -195,6 +195,9 @@ def test_causal_conv1d_update_with_batch_gather(
if not torch.cuda.is_available(): if not torch.cuda.is_available():
pytest.skip("CUDA device not available") pytest.skip("CUDA device not available")
# TODO: Fix Triton kernel bug - see issue #16714
pytest.skip("Triton kernel bug - see issue #16714")
device = "cuda" device = "cuda"
rtol, atol = (3e-4, 1e-3) if itype == torch.float32 else (3e-3, 5e-3) rtol, atol = (3e-4, 1e-3) if itype == torch.float32 else (3e-3, 5e-3)
if itype == torch.bfloat16: if itype == torch.bfloat16:
@@ -275,6 +278,9 @@ def test_causal_conv1d_varlen(
if not torch.cuda.is_available(): if not torch.cuda.is_available():
pytest.skip("CUDA device not available") pytest.skip("CUDA device not available")
# TODO: Fix Triton kernel bug - see issue #16714
pytest.skip("Triton kernel bug - see issue #16714")
device = "cuda" device = "cuda"
torch.cuda.empty_cache() torch.cuda.empty_cache()
rtol, atol = (3e-4, 1e-3) if itype == torch.float32 else (3e-3, 5e-3) rtol, atol = (3e-4, 1e-3) if itype == torch.float32 else (3e-3, 5e-3)