[amdgpu] LLVM 20 updates for AMD MI3xx GPUs

.github/workflows/scripts/ti_build/compiler.py

@@ -33,7 +33,7 @@ def setup_clang(as_compiler=True) -> None:
     """
     u = platform.uname()
     if u.system == "Linux":
-        for v in ("", "-14", "-13", "-12", "-11", "-10"):
+        for v in ("", "-20", "-19", "-18", "-17", "-16", "-15", "-14", "-13", "-12", "-11", "-10"):
             clang = shutil.which(f"clang{v}")
             if clang is not None:
                 clangpp = shutil.which(f"clang++{v}")

.github/workflows/scripts/ti_build/entry.py

@@ -82,8 +82,8 @@ def setup_basic_build_env():
         setup_clang()
 
     setup_llvm()
-    if u.system in ("Linux", "Windows"):
-        # We support & test Vulkan shader debug printf on Linux && Windows
+    if u.system == "Linux":
+        # We support & test Vulkan shader debug printf on Linux
         # This is done through the validation layer
         from .vulkan import setup_vulkan
 

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

@@ -1,3 +1,4 @@
+
 # -*- coding: utf-8 -*-
 
 # -- stdlib --
@@ -19,18 +20,25 @@ 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"
+            # We should use LLVM toolchains shipped with OS.
+            os.environ["LLVM_DIR"] = os.environ["LLVM_PATH"]+"/lib/cmake"
+            os.environ["CPATH"] = os.environ["ROCM_PATH"]+"/include"
         elif is_manylinux2014():
             # FIXME: prebuilt llvm15 on ubuntu didn't work on manylinux2014 image of centos. Once that's fixed, remove this hack.
             out = get_cache_home() / "llvm15-manylinux2014"
             url = "https://github.com/ailzhang/torchhub_example/releases/download/0.3/taichi-llvm-15-linux.zip"
+            download_dep(url, out, strip=1)
         else:
             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)
+            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"
@@ -47,5 +55,8 @@ def setup_llvm() -> None:
         raise RuntimeError(f"Unsupported platform: {u.system} {u.machine}")
 
     # We should use LLVM toolchains shipped with OS.
-    # path_prepend('PATH', out / 'bin')
-    os.environ["LLVM_DIR"] = str(out)
+     #path_prepend('PATH', out / 'bin')
+    if ((u.system, u.machine) not in (("Linux", "arm64"), ("Linux", "aarch64"))) and not (cmake_args.get_effective("TI_WITH_AMDGPU")):
+        os.environ["LLVM_DIR"] = "/usr/lib/llvm-20/cmake"
+        os.environ["CUDA_HOME"] = "/usr/local/cuda"
+        os.environ["CPATH"] = "/usr/local/cuda/include"