skip linear no igemmlt test

ROCm · May 9, 2024 · 06f6b25 · 06f6b25
1 parent 2af8568
commit 06f6b25
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_linear8bitlt.py b/tests/test_linear8bitlt.py
@@ -40,6 +40,7 @@ def test_layout_exact_match():
         assert torch.all(torch.eq(restored_x, x))
 
 
+@pytest.mark.skipif(HIP_ENVIRONMENT, reason="this test is not supported on ROCm yet")
 def test_linear_no_igemmlt():
     linear = torch.nn.Linear(1024, 3072)
     x = torch.randn(3, 1024, dtype=torch.half)