added checks to run only on CUDA with compatibility >=9

Author: avizon-aws

test/prototype/mx_formats/test_mxfp8_allgather.py

@@ -1,3 +1,4 @@
+import pytest
 import torch
 import torch.distributed as dist
 from torch.testing._internal.common_distributed import (
@@ -9,6 +10,11 @@
 
 from torchao.prototype.mx_formats.mx_tensor import MXTensor
 
+if not torch.cuda.is_available() or torch.cuda.get_device_capability() < (9, 0):
+    pytest.skip(
+        "Test Requires CUDA and compute capability >= 9.0", allow_module_level=True
+    )
+
 
 @instantiate_parametrized_tests
 class MXFP8OnDeviceAllGatherTest(MultiProcessTestCase):