From 85199f9681af6656c3f61f982e826f61664eb2af Mon Sep 17 00:00:00 2001 From: SoluMilken Date: Fri, 13 Mar 2026 00:08:37 +0800 Subject: [PATCH] [Bugfix] fix main branch pre-commit error (1 line change) (#36897) Signed-off-by: SoluMilken --- tests/kernels/test_fused_recurrent_packed_decode.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/kernels/test_fused_recurrent_packed_decode.py b/tests/kernels/test_fused_recurrent_packed_decode.py index f81f3c776..d63186bde 100644 --- a/tests/kernels/test_fused_recurrent_packed_decode.py +++ b/tests/kernels/test_fused_recurrent_packed_decode.py @@ -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(