skip test low bit optim

pytorch · Feb 5, 2025 · 01bab42 · 01bab42
1 parent 3168159
commit 01bab42
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/prototype/test_low_bit_optim.py b/test/prototype/test_low_bit_optim.py
@@ -43,6 +43,8 @@
 except ImportError:
     lpmm = None
 
+if torch.version.hip is not None:
+    pytest.skip("Skipping the test in ROCm", allow_module_level=True)
 
 _DEVICES = get_available_devices()