[CI] Add ROCM builds to the the nightly

Build Experimental ROCM builds
nod-ai · Jan 26, 2024 · 6d26a39 · 6d26a39
1 parent 405af4a
commit 6d26a39
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/compiler/setup.py b/compiler/setup.py
@@ -259,6 +259,8 @@ def prepare_installation():
             "-DPython3_EXECUTABLE={}".format(sys.executable),
             "-DCMAKE_BUILD_TYPE={}".format(cfg),
             # TODO(scotttodd): include IREE_TARGET_BACKEND_WEBGPU here (and in env)
+            get_env_cmake_option("IREE_TARGET_BACKEND_ROCM"),
+            get_env_cmake_option("IREE_TARGET_BACKEND_OPENCL_SPIRV"),
             get_env_cmake_option("IREE_ENABLE_CPUINFO", "ON"),
             get_env_cmake_option("IREE_TARGET_BACKEND_ROCM", "ON"),
             get_env_cmake_option("IREE_ENABLE_LLD", "OFF"),

diff --git a/runtime/setup.py b/runtime/setup.py
@@ -274,7 +274,8 @@ def build_configuration(cmake_build_dir, cmake_install_dir, extra_cmake_args=())
             "IREE_HAL_DRIVER_VULKAN",
             "OFF" if platform.system() == "Darwin" else "ON",
         ),
-        get_env_cmake_list("IREE_EXTERNAL_HAL_DRIVERS", ""),
+        get_env_cmake_list("IREE_EXTERNAL_HAL_DRIVERS",
+            "" if platform.system() != "Linux" else "rocm"),
         get_env_cmake_option("IREE_ENABLE_CPUINFO", "ON"),
     ] + list(extra_cmake_args)
     add_env_cmake_setting(cmake_args, "IREE_TRACING_PROVIDER")