[build] Upgrade Vulkan and Fix imgui

.github/workflows/scripts/ti_build/dep.py

@@ -153,6 +153,9 @@ def download_dep(url, outdir, *, strip=0, force=False, args=None, plain=False, e
     elif name.endswith(".tar.gz") or name.endswith(".tgz"):
         outdir.mkdir(parents=True, exist_ok=True)
         tar("-xzf", local_cached, "-C", outdir, f"--strip-components={strip}")
+    elif name.endswith(".tar.xz"):
+        outdir.mkdir(parents=True, exist_ok=True)
+        tar("-xJf", local_cached, "-C", outdir, f"--strip-components={strip}")
     elif name.endswith(".sh"):
         bash(local_cached, *args)
     elif "." not in name and args is not None:

.github/workflows/scripts/ti_build/llvm.py

@@ -19,7 +19,7 @@ def setup_llvm() -> None:
     Download and install LLVM.
     """
     u = platform.uname()
-    if u.system == "Linux":
+    if (u.system, u.machine) == ("Linux", "x86_64"):
         if cmake_args.get_effective("TI_WITH_AMDGPU"):
             out = get_cache_home() / "llvm15-amdgpu-005"
             url = "https://github.com/GaleSeLee/assets/releases/download/v0.0.5/taichi-llvm-15.0.0-linux.zip"
@@ -31,6 +31,11 @@ def setup_llvm() -> None:
             out = get_cache_home() / "llvm15"
             url = "https://github.com/taichi-dev/taichi_assets/releases/download/llvm15/taichi-llvm-15-linux.zip"
         download_dep(url, out, strip=1)
+
+    elif (u.system, u.machine) in (("Linux", "arm64"), ("Linux", "aarch64")):
+        out = get_cache_home() / "llvm15-manylinux2014"
+        # FIXME: ARM LLVM!
+        pass
     elif (u.system, u.machine) == ("Darwin", "arm64"):
         out = get_cache_home() / "llvm15-m1-nozstd"
         url = "https://github.com/taichi-dev/taichi_assets/releases/download/llvm15/taichi-llvm-15-m1-nozstd.zip"

.github/workflows/scripts/ti_build/ospkg.py

@@ -24,7 +24,7 @@
     "liblz4-dev",
     "libpng-dev",
     "libssl-dev",
-    "libtinfo-dev",
+    "libncurses-dev",
     "libwayland-dev",
     "libx11-xcb-dev",
     "libxcb-dri3-dev",

.github/workflows/scripts/ti_build/python.py

@@ -17,20 +17,23 @@
 
 
 # -- code --
-def setup_mambaforge(prefix):
+def setup_miniforge(prefix):
     u = platform.uname()
-    if u.system == "Linux":
-        url = "https://github.com/conda-forge/miniforge/releases/download/23.1.0-1/Mambaforge-23.1.0-1-Linux-x86_64.sh"
+    if (u.system, u.machine) == ("Linux", "x86_64"):
+        url = "https://github.com/conda-forge/miniforge/releases/download/25.3.0-1/Miniforge3-25.3.0-1-Linux-x86_64.sh"
+        download_dep(url, prefix, args=["-bfp", str(prefix)])
+    elif (u.system, u.machine) in (("Linux", "arm64"), ("Linux", "aarch64")):
+        url = "https://github.com/conda-forge/miniforge/releases/download/25.3.0-1/Miniforge3-25.3.0-1-Linux-aarch64.sh"
         download_dep(url, prefix, args=["-bfp", str(prefix)])
     elif (u.system, u.machine) == ("Darwin", "arm64"):
-        url = "https://github.com/conda-forge/miniforge/releases/download/23.1.0-1/Mambaforge-23.1.0-1-MacOSX-arm64.sh"
+        url = "https://github.com/conda-forge/miniforge/releases/download/25.3.0-1/Miniforge3-25.3.0-1-MacOSX-arm64.sh"
         download_dep(url, prefix, args=["-bfp", str(prefix)])
     elif (u.system, u.machine) == ("Darwin", "x86_64"):
-        url = "https://github.com/conda-forge/miniforge/releases/download/23.1.0-1/Mambaforge-23.1.0-1-MacOSX-x86_64.sh"
+        url = "https://github.com/conda-forge/miniforge/releases/download/25.3.0-1/Miniforge3-25.3.0-1-MacOSX-x86_64.sh"
         download_dep(url, prefix, args=["-bfp", str(prefix)])
     elif u.system == "Windows":
         url = (
-            "https://github.com/conda-forge/miniforge/releases/download/23.1.0-1/Mambaforge-23.1.0-1-Windows-x86_64.exe"
+            "https://github.com/conda-forge/miniforge/releases/download/25.3.0-1/Miniforge3-25.3.0-1-Windows-x86_64.exe"
         )
         download_dep(
             url,
@@ -82,8 +85,8 @@ def setup_python(version: str) -> Tuple[Command, Command]:
 
     windows = platform.system() == "Windows"
 
-    prefix = get_cache_home() / "mambaforge"
-    setup_mambaforge(prefix)
+    prefix = get_cache_home() / "miniforge"
+    setup_miniforge(prefix)
 
     if windows:
         conda_path = prefix / "Scripts" / "conda.exe"
@@ -92,9 +95,9 @@ def setup_python(version: str) -> Tuple[Command, Command]:
 
     if not conda_path.exists():
         shutil.rmtree(prefix, ignore_errors=True)
-        setup_mambaforge(prefix)
+        setup_miniforge(prefix)
         if not conda_path.exists():
-            raise RuntimeError(f"Failed to setup mambaforge at {prefix}")
+            raise RuntimeError(f"Failed to setup miniforge at {prefix}")
 
     conda = sh.bake(str(conda_path))
 

.github/workflows/scripts/ti_build/sccache.py

@@ -31,18 +31,16 @@ def setup_sccache() -> Command:
         raise RuntimeError(f"Unsupported platform: {u.system} {u.machine}")
 
     if not exe.exists():
-        if u.system == "Linux":
-            url = "https://github.com/mozilla/sccache/releases/download/v0.4.1/sccache-v0.4.1-x86_64-unknown-linux-musl.tar.gz"
+        if (u.system, u.machine) == ("Linux", "x86_64"):
+            url = "https://github.com/mozilla/sccache/releases/download/v0.10.0/sccache-v0.10.0-x86_64-unknown-linux-musl.tar.gz"
+        elif (u.system, u.machine) in (("Linux", "arm64"), ("Linux", "aarch64")):
+            url = "https://github.com/mozilla/sccache/releases/download/v0.10.0/sccache-v0.10.0-aarch64-unknown-linux-musl.tar.gz"
         elif (u.system, u.machine) == ("Darwin", "arm64"):
-            url = (
-                "https://github.com/mozilla/sccache/releases/download/v0.4.1/sccache-v0.4.1-aarch64-apple-darwin.tar.gz"
-            )
+            url = "https://github.com/mozilla/sccache/releases/download/v0.10.0/sccache-v0.10.0-aarch64-apple-darwin.tar.gz"
         elif (u.system, u.machine) == ("Darwin", "x86_64"):
-            url = (
-                "https://github.com/mozilla/sccache/releases/download/v0.4.1/sccache-v0.4.1-x86_64-apple-darwin.tar.gz"
-            )
+            url = "https://github.com/mozilla/sccache/releases/download/v0.10.0/sccache-v0.10.0-x86_64-apple-darwin.tar.gz"
         elif u.system == "Windows":
-            url = "https://github.com/mozilla/sccache/releases/download/v0.4.1/sccache-v0.4.1-x86_64-pc-windows-msvc.tar.gz"
+            url = "https://github.com/mozilla/sccache/releases/download/v0.10.0/sccache-v0.10.0-x86_64-pc-windows-msvc.tar.gz"
         else:
             raise RuntimeError(f"Unsupported platform: {u.system} {u.machine}")
 

.github/workflows/scripts/ti_build/vulkan.py

@@ -1,3 +1,4 @@
+
 # -*- coding: utf-8 -*-
 
 # -- stdlib --
@@ -12,23 +13,32 @@
 
 
 # -- code --
-@banner("Setup Vulkan 1.3.236.0")
+@banner("Setup Vulkan 1.4.313.0")
 def setup_vulkan():
     u = platform.uname()
     if u.system == "Linux":
-        url = "https://sdk.lunarg.com/sdk/download/1.3.236.0/linux/vulkansdk-linux-x86_64-1.3.236.0.tar.gz"
-        prefix = get_cache_home() / "vulkan-1.3.236.0"
+        url = "https://sdk.lunarg.com/sdk/download/1.4.313.0/linux/vulkansdk-linux-x86_64-1.4.313.0.tar.xz"
+        prefix = get_cache_home() / "vulkan-1.4.313.0"
         download_dep(url, prefix, strip=1)
         sdk = prefix / "x86_64"
         os.environ["VULKAN_SDK"] = str(sdk)
         path_prepend("PATH", sdk / "bin")
         path_prepend("LD_LIBRARY_PATH", sdk / "lib")
         os.environ["VK_LAYER_PATH"] = str(sdk / "etc" / "vulkan" / "explicit_layer.d")
+    elif (u.system, u.machine) in (("Linux", "arm64"), ("Linux", "aarch64")):
+        url = "https://github.com/johnnynunez/vulkan-sdk-arm/releases/download/1.4.313.0/vulkansdk-linux-arm64-ubuntu-24.04-arm-1.4.313.0.zip"
+        prefix = get_cache_home() / "vulkan-1.4.313.0"
+        download_dep(url, prefix, strip=1)
+        sdk = prefix / "arm64"
+        os.environ["VULKAN_SDK"] = str(sdk)
+        path_prepend("PATH", sdk / "bin")
+        path_prepend("LD_LIBRARY_PATH", sdk / "lib")
+        os.environ["VK_LAYER_PATH"] = str(sdk / "etc" / "vulkan" / "explicit_layer.d")
     # elif (u.system, u.machine) == ("Darwin", "arm64"):
     # elif (u.system, u.machine) == ("Darwin", "x86_64"):
     elif (u.system, u.machine) == ("Windows", "AMD64"):
-        url = "https://sdk.lunarg.com/sdk/download/1.3.236.0/windows/VulkanSDK-1.3.236.0-Installer.exe"
-        prefix = get_cache_home() / "vulkan-1.3.236.0"
+        url = "https://sdk.lunarg.com/sdk/download/1.4.313.0/windows/VulkanSDK-1.4.313.0-Installer.exe"
+        prefix = get_cache_home() / "vulkan-1.4.313.0"
         download_dep(
             url,
             prefix,
@@ -53,3 +63,4 @@ def setup_vulkan():
         path_prepend("PATH", prefix / "Bin")
     else:
         return
+

python/taichi/_version_check.py

@@ -18,8 +18,11 @@ def check_version(cur_uuid):
     payload = {"version": version, "platform": "", "python": ""}
 
     system = platform.system()
-    if system == "Linux":
+    u = platform.uname()
+    if (u.system, u.machine) == ("Linux", "x86_64"):
         payload["platform"] = "manylinux_2_27_x86_64"
+    elif (u.system, u.machine) in (("Linux", "arm64"), ("Linux", "aarch64")):
+        payload["platform"] = "manylinux_2_27_aarch64"
     elif system == "Windows":
         payload["platform"] = "win_amd64"
     elif system == "Darwin":

taichi/analysis/verify.cpp

@@ -97,7 +97,7 @@ class IRVerifier : public BasicStmtVisitor {
     } else if (!stmt->has_body() && stmt->body) {
       TI_ERROR("offloaded {} ({})->body is {} (should be nullptr)",
                offloaded_task_type_name(stmt->task_type), stmt->name(),
-               fmt::ptr(stmt->body));
+               fmt::ptr(stmt->body.get()));
     }
     stmt->all_blocks_accept(this);
   }

taichi/common/logging.h

@@ -10,6 +10,7 @@
 // before including "spdlog/fmt/fmt.h"
 #include "spdlog/common.h"
 #include "spdlog/fmt/fmt.h"
+#include "spdlog/fmt/ranges.h"
 namespace spdlog {
 class logger;
 }

taichi/ui/ggui/gui.cpp

@@ -44,7 +44,9 @@ Gui::Gui(AppContext *app_context, SwapChain *swap_chain, TaichiWindow *window) {
 
 void Gui::init_render_resources(VkRenderPass render_pass) {
   ImGui_ImplVulkan_LoadFunctions(
-      load_vk_function_for_gui);  // this is because we're using volk.
+      VK_API_VERSION_1_0,        // or app_context_->config.vk_api_version
+      load_vk_function_for_gui,  // this is because we're using volk.
+      nullptr);
 
   auto &device =
       static_cast<taichi::lang::vulkan::VulkanDevice &>(app_context_->device());
@@ -60,7 +62,9 @@ void Gui::init_render_resources(VkRenderPass render_pass) {
   init_info.Allocator = VK_NULL_HANDLE;
   init_info.MinImageCount = swap_chain_->surface().get_image_count();
   init_info.ImageCount = swap_chain_->surface().get_image_count();
-  ImGui_ImplVulkan_Init(&init_info, render_pass);
+  // new signature takes only the struct
+  init_info.RenderPass = render_pass;
+  ImGui_ImplVulkan_Init(&init_info);
   render_pass_ = render_pass;
 
   // Upload Fonts
@@ -73,10 +77,11 @@ void Gui::init_render_resources(VkRenderPass render_pass) {
             ->vk_command_buffer()
             ->buffer;
 
-    ImGui_ImplVulkan_CreateFontsTexture(command_buffer);
+    // ≥ 1.90: the helper records its own commands
+    ImGui_ImplVulkan_CreateFontsTexture();
 
     stream->submit_synced(cmd_list.get());
-    ImGui_ImplVulkan_DestroyFontUploadObjects();
+    ImGui_ImplVulkan_DestroyFontsTexture();
   }
 
   prepare_for_next_frame();