Skip to content

Commit ca78ddd

Browse files
Tests: skip test_gemv_eye_4bit on CPU with bf16 when not supported by torch
1 parent 4ee6c60 commit ca78ddd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_functional.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1330,7 +1330,7 @@ def test_gemv_4bit(self, device, dim, dtype, storage_type, quant_storage, double
13301330
@pytest.mark.parametrize("dtype", [torch.float16, torch.bfloat16, torch.float32], ids=describe_dtype)
13311331
@pytest.mark.parametrize("double_quant", [False], ids=["DQ_True"])
13321332
def test_gemv_eye_4bit(self, device, storage_type, dtype, double_quant):
1333-
if dtype == torch.bfloat16 and torch.__version__ < (2, 3):
1333+
if device == "cpu" and dtype == torch.bfloat16 and torch.__version__ < (2, 3):
13341334
pytest.skip("eye doe not support bfloat16 on CPU in torch < 2.3")
13351335

13361336
dims = 10

0 commit comments

Comments
 (0)