[Bugfix] fix main branch pre-commit error (1 line change) (#36897)

Signed-off-by: SoluMilken <ypiheyn.imm02g@g2.nctu.edu.tw>
This commit is contained in:
SoluMilken
2026-03-13 00:08:37 +08:00
committed by GitHub
parent a1257fd1ea
commit 85199f9681

View File

@@ -10,7 +10,7 @@ from vllm.model_executor.layers.fla.ops import (
)
@pytest.mark.skipif(torch.cuda.device_count() < 1, reason="Need CUDA device")
@pytest.mark.skipif(not torch.cuda.is_available(), reason="Need CUDA device")
@pytest.mark.parametrize("dtype", [torch.float16, torch.bfloat16, torch.float32])
@pytest.mark.parametrize("strided_mixed_qkv", [False, True])
def test_fused_recurrent_packed_decode_matches_reference(