Skip to content

Commit 55da7f3

Browse files
skip gemv tests on hpu
1 parent bdd28f2 commit 55da7f3

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

tests/test_functional.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1211,8 +1211,8 @@ def test_bench_4bit_dequant(self, quant_type):
12111211
)
12121212
@pytest.mark.parametrize("dim", [128, 256, 512, 1024], ids=id_formatter("dim"))
12131213
def test_gemv_4bit(self, device, dim, dtype, storage_type, quant_storage, double_quant, kind):
1214-
if device == "hpu" and storage_type != "nf4":
1215-
pytest.skip("fp4 dequantization is not supported on HPU")
1214+
if device == "hpu":
1215+
pytest.skip("gemv not supported on HPU")
12161216

12171217
errs1 = []
12181218
errs2 = []

tests/test_ops.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -213,8 +213,8 @@ def test_dequantize_4bit(self, device, dtype, storage_dtype, quant_type, blocksi
213213
@pytest.mark.parametrize("quant_type", ["fp4", "nf4"])
214214
@pytest.mark.parametrize("blocksize", [64, 128, 256, 512])
215215
def test_gemv_4bit(self, device, dtype, storage_dtype, quant_type, blocksize):
216-
if device == "hpu" and quant_type != "nf4":
217-
pytest.skip("fp4 dequantization is not supported on HPU")
216+
if device == "hpu":
217+
pytest.skip("gemv not supported on HPU")
218218

219219
out_features = 1024
220220
in_features = 256

0 commit comments

Comments
 (0)