Skip overflow and linear serialization for now

ROCm · Mar 12, 2024 · 90bbdc6 · 90bbdc6
1 parent 93dfb51
commit 90bbdc6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tests/test_functional.py b/tests/test_functional.py
@@ -1356,6 +1356,7 @@ def test_transform(dim1, dim2, dim3, dims, dtype, orderA, orderOut, transpose):
     for vals in values
 ]
 
+@pytest.mark.skipif(HIP_ENVIRONMENT, reason="this test is not supported on ROCm yet")
 def test_overflow():
     formatB = F.get_special_format_str()
     print(formatB)

diff --git a/tests/test_linear8bitlt.py b/tests/test_linear8bitlt.py
@@ -68,6 +68,7 @@ def test_linear_no_igemmlt():
     assert linear_custom.state.CxB is None
 
 
+@pytest.mark.skipif(HIP_ENVIRONMENT, reason="this test is not supported on ROCm yet")
 @pytest.mark.parametrize("has_fp16_weights, serialize_before_forward, deserialize_before_cuda, force_no_igemmlt",
                          list(product([False, True], [False, True], [False, True], [False, True])))
 def test_linear_serialization(has_fp16_weights, serialize_before_forward, deserialize_before_cuda, force_no_igemmlt):