add GPU arch check for MI300x

Author: Peter Y. Yeh

setup.py

@@ -146,6 +146,12 @@ def get_extensions():
 
     # TOOD: Remove this and use what CUDA has once we fix all the builds.
     if IS_ROCM and use_cuda:
+        # Add ROCm GPU architecture check
+        gpu_arch = torch.cuda.get_device_properties(0).name
+        if gpu_arch != 'gfx942':
+            print(f"Warning: Unsupported ROCm GPU architecture: {gpu_arch}")
+            print("Currently only gfx942 is supported. Skipping compilation of ROCm extensions")
+            return None
         sources += hip_sources
 
     if len(sources) == 0: