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

Fix issues/16714: Revert comment out of tl.debug_barrier() in causal_conv1d_triton (#16899)

Browse Source 
Co-authored-by: Yi Zhang <1109276519@qq.com>
This commit is contained in:
Netanel Haber
2026-01-14 17:26:48 +08:00
committed by GitHub
co-authored by Yi Zhang
parent 72bacc88c8
commit e75299a111
2 changed files with 3 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/layers/mamba/test_causal_conv1d.py
-6
View File
@@ -196,9 +196,6 @@ def test_causal_conv1d_update_with_batch_gather(
if not torch.cuda.is_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"
rtol, atol = (3e-4, 1e-3) if itype == torch.float32 else (3e-3, 5e-3)
if itype == torch.bfloat16:
@@ -279,9 +276,6 @@ def test_causal_conv1d_varlen(
if not torch.cuda.is_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"
torch.cuda.empty_cache()
rtol, atol = (3e-4, 1e-3) if itype == torch.float32 else (3e-3, 5e-3)