fixing test

Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags:
pytorch · Oct 25, 2024 · 4f45e03 · 4f45e03
1 parent d3af4d5
commit 4f45e03
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/integration/test_integration.py b/test/integration/test_integration.py
@@ -1297,6 +1297,8 @@ def test_autoquant_compile(self, device, dtype, m1, m2, k, n):
     @parameterized.expand(COMMON_DEVICE_DTYPE)
     @unittest.skipIf(not TORCH_VERSION_AT_LEAST_2_5, "autoquant requires 2.5+.")
     def test_autoquant_mha(self, device, dtype):
+        if device != "cuda" or not torch.cuda.is_available():
+            self.skipTest(f"autoquant currently does not support {device}")
         class MHAModel(torch.nn.Module):
             def __init__(self):
                 super().__init__()