test(tinygemm): skip mps device until pt 2.4

huggingface · Jul 11, 2024 · ae05a6e · ae05a6e
1 parent df5e7e7
commit ae05a6e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/tensor/qbits/test_tinygemm_qbits_tensor.py b/test/tensor/qbits/test_tinygemm_qbits_tensor.py
@@ -56,6 +56,7 @@ def test_tinygemm_qbits_tensor_from_qbits_tensor(in_features, out_features, devi
     assert_similar(tgqbt.dequantize(), qbt.dequantize())
 
 
+@pytest.mark.skip_device("mps")  # Only available with pytorch 2.4
 def test_tinygemm_qbits_tensor_move(device):
     qtype = qint4
     group_size = 128