fixing tests

Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags:
pytorch · Aug 7, 2024 · 56b44aa · 56b44aa
1 parent 2623bab
commit 56b44aa
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/test/integration/test_integration.py b/test/integration/test_integration.py
@@ -672,6 +672,8 @@ def _test_lin_weight_subclass_impl(
         test_dtype=torch.bfloat16,
         test_shape=(32, 64, 32),
     ):
+        if not "cuda" in test_device:
+            self.skipTest("test requires cuda")
         m, k, n = test_shape
         x = torch.randn(m, k, device=test_device, dtype=test_dtype)
         lin = torch.nn.Linear(k, n, device=test_device).to(test_dtype)
@@ -714,12 +716,6 @@ def test_aq_int8_dynamic_quant_subclass(self, device, dtype):
             AQInt8DynamicallyQuantizedLinearWeight.from_float, device, 35, test_dtype=dtype
         )
 
-    @parameterized.expand(COMMON_DEVICE_DTYPE)
-    def test_aq_int8_weight_only_quant_subclass(self, device, dtype):
-        self._test_lin_weight_subclass_impl(
-            AQInt8DynamicallyQuantizedLinearWeight.from_float, device, 35, test_dtype=dtype
-        )
-
     @parameterized.expand(COMMON_DEVICE_DTYPE)
     def test_aq_int8_weight_only_quant_subclass(self, device, dtype):
         self._test_lin_weight_subclass_impl(