Merge pull request #2048 from rhatdan/fix-issue-2045

Fix AMD GPU image selection on arm64 for issue #2045

Author: rhatdan

ramalama/common.py

@@ -433,6 +433,10 @@ def check_ascend() -> Literal["cann"] | None:
 
 
 def check_rocm_amd() -> Literal["hip"] | None:
+    if is_arm():
+        # ROCm is not available for arm64, use Vulkan instead
+        return None
+
     gpu_num = 0
     gpu_bytes = 0
     for i, (np, props) in enumerate(amdkfd.gpus()):
@@ -461,6 +465,10 @@ def check_rocm_amd() -> Literal["hip"] | None:
     return None
 
 
+def is_arm() -> bool:
+    return platform.machine() in ('arm64', 'aarch64')
+
+
 def check_intel() -> Literal["intel"] | None:
     igpu_num = 0
     # Device IDs for select Intel GPUs.  See: https://dgpu-docs.intel.com/devices/hardware-table.html