fix aarch

taichi-dev · Feb 2, 2025 · 8591d9a · 8591d9a
1 parent 70679b8
commit 8591d9a
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/python/taichi/_version_check.py b/python/taichi/_version_check.py
@@ -17,17 +17,14 @@ def check_version(cur_uuid):
     version = f"{major}.{minor}.{patch}"
     payload = {"version": version, "platform": "", "python": ""}
 
-    system, machine = platform.uname().system, platform.uname().machine
-    if system == "Linux":
-        if machine == "x86_64":
-            payload["platform"] = "manylinux_2_28_x86_64"
-        else:
-            payload["platform"] = "manylinux_2_28_aarch64"
+    system = platform.system()
+    u = platform.uname()
+    if (u.system, u.machine) == ("Linux", "x86_64"):
+        payload["platform"] = "manylinux_2_28_x86_64"
+    elif (u.system, u.machine) in (("Linux", "arm64"), ("Linux", "aarch64")):
+        payload["platform"] = "manylinux_2_28_aarch64"
     elif system == "Windows":
-        if machine == "AMD64":
-            payload["platform"] = "win_amd64"
-        else:
-            payload["platform"] = "win_arm64"
+        payload["platform"] = "win_amd64"
     elif system == "Darwin":
         if platform.release() < "19.0.0":
             payload["platform"] = "macosx_10_14_x86_64"