Fix ROCm GPU architecture detection in setup.py

Update GPU architecture check to use gcnArchName and improve detection of gfx942 support

Author: petrex

setup.py

@@ -363,8 +363,8 @@ 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":
+        gpu_arch = torch.cuda.get_device_properties(0).gcnArchName
+        if "gfx942" not in gpu_arch:
             print(f"Warning: Unsupported ROCm GPU architecture: {gpu_arch}")
             print(
                 "Currently only gfx942 is supported. Skipping compilation of ROCm extensions"