Fix CI float8 test

jainapurva · jainapurva · commit 08bccdc13c09 · 2025-01-17T14:43:16.000-08:00
diff --git a/test/integration/test_integration.py b/test/integration/test_integration.py
@@ -1821,7 +1821,7 @@ def test_autoquant_int4wo(self, device, dtype):
             self.assertGreater(compute_error(ref, out), 20)
 
     @parameterized.expand(COMMON_DEVICE_DTYPE)
-    @unittest.skipIf(not torch.cuda.is_available(), "Need CUDA available")
+    @unittest.skipIf(not is_sm_at_least_90(), "Need cuda arch greater than SM90")
     @unittest.skipIf(
         not TORCH_VERSION_AT_LEAST_2_5, "autoquant int4 option requires 2.5+."
     )

Original file line number	Diff line number	Diff line change
`@@ -1821,7 +1821,7 @@ def test_autoquant_int4wo(self, device, dtype):`
`1821`	`1821`	`self.assertGreater(compute_error(ref, out), 20)`
`1822`	`1822`
`1823`	`1823`	`@parameterized.expand(COMMON_DEVICE_DTYPE)`
`1824`		`- @unittest.skipIf(not torch.cuda.is_available(), "Need CUDA available")`
	`1824`	`+ @unittest.skipIf(not is_sm_at_least_90(), "Need cuda arch greater than SM90")`
`1825`	`1825`	`@unittest.skipIf(`
`1826`	`1826`	`not TORCH_VERSION_AT_LEAST_2_5, "autoquant int4 option requires 2.5+."`
`1827`	`1827`	`)`