diff --git a/compiler/setup.py b/compiler/setup.py index 01a347d296b97..4ff0ddf68f046 100644 --- 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 index e561c45ea6e5d..345f31fc92914 100644 --- 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")