Update test_bits.py

pytorch · Dec 5, 2023 · 157c7a5 · 157c7a5
1 parent 7aa06b3
commit 157c7a5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/quantization/core/experimental/test_bits.py b/test/quantization/core/experimental/test_bits.py
@@ -3,7 +3,7 @@
 import torch
 from torch.testing._internal.common_device_type import instantiate_device_type_tests
 
-from torch.testing._internal.common_utils import run_tests, TestCase
+from torch.testing._internal.common_utils import run_tests, TestCase, skipIfRocm
 from torch.utils._mode_utils import no_dispatch
 from torch.utils._pytree import tree_map
 
@@ -45,6 +45,7 @@ def __repr__(self) -> str:
 
 
 class TestBits(TestCase):
+    @skipIfRocm
     def test_types(self, device):
         bits_types = [torch.bits1x8, torch.bits2x4, torch.bits4x2, torch.bits8, torch.bits16]
         for bits_type in bits_types: