Skips for ROCm (X86 Inductor Tests) (#2202)

* Refactor ROCm skip decorators in quantization tests

Updated the skip decorators for ROCm in the quantization test suite to use the new `skip_if_rocm` utility, providing more descriptive messages for each test case. This change enhances clarity regarding ROCm support status in the tests.

* Reorganize ROCm skip decorator import in quantization tests

Moved the import statement for `skip_if_rocm` to maintain consistency in the test file structure. This change ensures that the utility is properly utilized in the context of the x86 Inductor quantization tests.

* Update ROCm skip decorator message in quantization test

Changed the message for the `skip_if_rocm` decorator in the `test_qat_qconv2d` test case to clarify that ROCm support is not applicable, enhancing the clarity of the test's intent.

Author: petrex

test/quantization/pt2e/test_x86inductor_fusion.py

@@ -28,7 +28,6 @@
     IS_X86,
     instantiate_parametrized_tests,
     parametrize,
-    skipIfRocm,
 )
 from torch.testing._internal.inductor_utils import (
     HAS_CPU,
@@ -45,6 +44,7 @@
 from torchao.quantization.pt2e.quantizer.x86_inductor_quantizer import (
     X86InductorQuantizer,
 )
+from torchao.testing.utils import skip_if_rocm
 from torchao.utils import (
     TORCH_VERSION_AT_LEAST_2_6,
     TORCH_VERSION_AT_LEAST_2_8,
@@ -328,7 +328,7 @@ def matcher_check_fn():
 
     @skipIfNoDynamoSupport
     @skipIfNoONEDNN
-    @skipIfRocm
+    @skip_if_rocm("Not applicable to ROCm")
     def test_qconv2d_cpu(self):
         r"""
         This testcase will quantize a single Conv2d module.
@@ -338,7 +338,7 @@ def test_qconv2d_cpu(self):
     @skipIfNoDynamoSupport
     @skipIfNoONEDNNBF16
     @skipIfNoONEDNN
-    @skipIfRocm
+    @skip_if_rocm("Not applicable to ROCm")
     def test_qconv2d_int8_mixed_bf16(self):
         r"""
         This testcase will quantize a single Conv2d module with int8_mixed_bf16 quantization.
@@ -932,7 +932,7 @@ def matcher_check_fn():
 
     @skipIfNoDynamoSupport
     @skipIfNoONEDNN
-    @skipIfRocm
+    @skip_if_rocm("Not applicable to ROCm")
     def test_qat_qconv2d(self):
         r"""
         This testcase will quantize a single Conv2d module with qat flow.
@@ -1075,7 +1075,7 @@ def test_qat_qconv2d_hardswish(self):
 
     @skipIfNoDynamoSupport
     @skipIfNoONEDNN
-    @skipIfRocm
+    @skip_if_rocm("Not applicable to ROCm")
     def test_qat_qconv2d_add(self):
         r"""
         This testcase will quantize a Conv2d->Add pattern as:
@@ -1141,7 +1141,7 @@ def matcher_check_fn():
 
     @skipIfNoDynamoSupport
     @skipIfNoONEDNN
-    @skipIfRocm
+    @skip_if_rocm("Not applicable to ROCm")
     def test_qat_qconv2d_add_relu(self):
         r"""
         This testcase will quantize a Conv2d->Add->ReLU pattern as:
@@ -1281,7 +1281,7 @@ def matcher_check_fn():
 
     @skipIfNoDynamoSupport
     @skipIfNoONEDNN
-    @skipIfRocm
+    @skip_if_rocm("Not applicable to ROCm")
     def test_qconv2d_dequant_promotion_cpu(self):
         self._test_qconv2d_dequant_promotion_helper()