Skip Unit Tests for ROCm CI

[petrex]

Skip the unit tests for ROCm CI deployment. refer to #999

This pull request introduces a decorator to skip tests on the ROCm platform and applies this decorator across multiple test files. The most important changes include the addition of the skip_if_rocm decorator and its application to various test functions to handle ROCm-specific issues.

Introduction of skip_if_rocm decorator:

• test/test_utils.py: Added the skip_if_rocm decorator to skip tests on the ROCm platform with a custom message.

Application of skip_if_rocm decorator:

• test/dtypes/test_affine_quantized.py: Applied @skip_if_rocm to multiple test functions and imports. [1] [2] [3] [4]
• test/dtypes/test_floatx.py: Applied @skip_if_rocm to test functions and imports. [1] [2]
• test/float8/test_base.py: Applied @skip_if_rocm to test functions and imports. [1] [2]
• test/hqq/test_hqq_affine.py: Applied @skip_if_rocm to test functions and imports. [1] [2]
• test/integration/test_integration.py: Applied @skip_if_rocm to multiple test functions and imports. [1] [2] [3] [4] [5] [6]
• test/kernel/test_galore_downproj.py: Applied @skip_if_rocm to test functions and imports. [1] [2]
• test/prototype/test_awq.py: Applied @skip_if_rocm to test functions and imports. [1] [2]
• test/prototype/test_low_bit_optim.py: Applied @skip_if_rocm to test functions and imports. [1] [2]
• test/prototype/test_splitk.py: Applied @skip_if_rocm to test functions and imports. (test/prototype/test_splitk.pyL17-R23, F6c2ed6dL19R19)
• test/quantization/test_galore_quant.py: Applied @skip_if_rocm to test functions and imports. [1] [2]
• test/quantization/test_marlin_qqq.py: Applied @skip_if_rocm to test functions and imports. [1] [2] [3]
• test/sparsity/test_marlin.py: Applied @skip_if_rocm to test functions and imports. [1] [2] [3]
• test/test_ops.py: Added a module-level skip for ROCm.
• test/test_s8s4_linear_cutlass.py: Added a module-level skip for ROCm.

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/1563

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

⏳ No Failures, 2 Pending

As of commit 38cab14 with merge base b3deb16 ():
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[pruthvistony]

"ROCm enablement in progress" would be a better comment.

[pruthvistony]

ROCm CI enable-ment in progress. So skipping few UTs temporary, to be debugged and fixed in further PRs.

[andrewor14]

This caused CI to fail:

  /opt/conda/envs/venv/lib/python3.9/importlib/__init__.py:127: in import_module
      return _bootstrap._gcd_import(name[level:], package, level)
  test/quantization/test_marlin_qqq.py:6: in <module>
      from test_utils import skip_if_rocm
  E   ModuleNotFoundError: No module named 'test_utils'

I'm reverting this for now