Skip Flaky ROCm Test in Integration Suite (#2691)

petrex · web-flow · commit 3c2e229b36c9 · 2025-08-05T10:02:42.000-07:00
Add skip condition for flaky ROCm test in integration suite
diff --git a/test/integration/test_integration.py b/test/integration/test_integration.py
@@ -1058,6 +1058,7 @@ def test_int8_weight_only_quant_subclass_api(self, device, dtype):
     @unittest.skipIf(
         not TORCH_VERSION_AT_LEAST_2_4, "freeze requires torch 2.4 and after."
     )
+    @skip_if_rocm("Test flaky on ROCm, under investigation")
     def test_int8_weight_only_quant_with_freeze(self, device, dtype):
         torch._dynamo.reset()
         self._test_lin_weight_subclass_api_impl(

Original file line number	Diff line number	Diff line change
`@@ -1058,6 +1058,7 @@ def test_int8_weight_only_quant_subclass_api(self, device, dtype):`
`1058`	`1058`	`@unittest.skipIf(`
`1059`	`1059`	`not TORCH_VERSION_AT_LEAST_2_4, "freeze requires torch 2.4 and after."`
`1060`	`1060`	`)`
	`1061`	`+ @skip_if_rocm("Test flaky on ROCm, under investigation")`
`1061`	`1062`	`def test_int8_weight_only_quant_with_freeze(self, device, dtype):`
`1062`	`1063`	`torch._dynamo.reset()`
`1063`	`1064`	`self._test_lin_weight_subclass_api_impl(`