also return aarch64 on Mac

rapidsai · Oct 9, 2024 · 7ef77c7 · 7ef77c7
1 parent 2bf78f1
commit 7ef77c7
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/rapids_build_backend/impls.py b/rapids_build_backend/impls.py
@@ -53,15 +53,20 @@ def _get_backend(build_backend):
 
 
 @lru_cache
-def _get_arch():
+def _get_arch() -> str:
     """Get the arch of the current machine.
 
     Returns
     -------
     str
         The arch (e.g. "x86_64" or "aarch64")
     """
-    return platform.machine()
+    plat = platform.machine()
+    # RAPIDS projects all use "aarch64" to indicate arm architectures,
+    # but arm some systems (like the M1/M2/M3 Macs) report "arm64"
+    if plat == "arm64":
+        return "aarch64"
+    return plat
 
 
 @lru_cache