"vscode:/vscode.git/clone" did not exist on "ada4f4fadd20372b1bf349961a1e442b2d07c53d"
Unverified Commit 85199f96 authored by SoluMilken's avatar SoluMilken Committed by GitHub
Browse files

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


Signed-off-by: default avatarSoluMilken <ypiheyn.imm02g@g2.nctu.edu.tw>
parent a1257fd1
......@@ -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(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment