Skip to content

Commit ef9e03c

Browse files
committed
Skip gemv 4 bit cpu test
1 parent cd026c3 commit ef9e03c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/test_functional.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2303,6 +2303,7 @@ def test_gemv_4bit(dtype, storage_type, quant_storage, double_quant, kind):
23032303
assert maxratio < 1.02 and maxratio > 0.98
23042304

23052305

2306+
@pytest.mark.skipif(HIP_ENVIRONMENT, reason="this test is not supported on ROCm yet")
23062307
@pytest.mark.parametrize("kind", ["fc1", "fc2", "attn", "attn_packed"])
23072308
@pytest.mark.parametrize("quant_type", ["nf4", "fp4"])
23082309
@pytest.mark.parametrize("dtype", [torch.float16, torch.bfloat16, torch.float32], ids=describe_dtype)

0 commit comments

Comments
 (0)