This is an automated email from the ASF dual-hosted git repository.
mshr pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 8c259a38a1 [REFACTOR][CI] Further cleanup docker images (#18817)
8c259a38a1 is described below
commit 8c259a38a1dfa84f6b8f17e6d9e3bde5185365ff
Author: Masahiro Hiramori <[email protected]>
AuthorDate: Thu Feb 26 11:23:13 2026 +0900
[REFACTOR][CI] Further cleanup docker images (#18817)
This PR removes unused legacy installer scripts from CI images.
- Removed packages: Arm Compute Library (acl), Hexagon SDK tooling,
libxsmm, papi, redis, gluoncv, and wasmtime.
---
docker/Dockerfile.ci_arm | 12 ++--
docker/Dockerfile.ci_cpu | 21 +------
docker/Dockerfile.ci_gpu | 15 ++---
.../ubuntu_download_arm_compute_lib_binaries.sh | 66 ----------------------
docker/install/ubuntu_install_hexagon.sh | 47 ---------------
docker/install/ubuntu_install_libxsmm.sh | 32 -----------
docker/install/ubuntu_install_papi.sh | 35 ------------
docker/install/ubuntu_install_redis.sh | 25 --------
docker/install/ubuntu_install_wasmtime.sh | 53 -----------------
...nstall_gluoncv.sh => ubuntu_install_xgboost.sh} | 2 +-
10 files changed, 12 insertions(+), 296 deletions(-)
diff --git a/docker/Dockerfile.ci_arm b/docker/Dockerfile.ci_arm
index b46bcd4cf5..7c235b0386 100644
--- a/docker/Dockerfile.ci_arm
+++ b/docker/Dockerfile.ci_arm
@@ -70,14 +70,10 @@ RUN bash /install/ubuntu_install_tensorflow_aarch64.sh
COPY install/ubuntu_install_tflite.sh /install/ubuntu_install_tflite.sh
RUN bash /install/ubuntu_install_tflite.sh
-# ONNX
+# ONNX and PyTorch
COPY install/ubuntu_install_onnx.sh /install/ubuntu_install_onnx.sh
RUN bash /install/ubuntu_install_onnx.sh
-# AutoTVM deps
-COPY install/ubuntu_install_redis.sh /install/ubuntu_install_redis.sh
-RUN bash /install/ubuntu_install_redis.sh
-
-# Arm(R) Compute Library
-COPY install/ubuntu_download_arm_compute_lib_binaries.sh
/install/ubuntu_download_arm_compute_lib_binaries.sh
-RUN bash /install/ubuntu_download_arm_compute_lib_binaries.sh
+# MetaScheduler deps
+COPY install/ubuntu_install_xgboost.sh /install/ubuntu_install_xgboost.sh
+RUN bash /install/ubuntu_install_xgboost.sh
diff --git a/docker/Dockerfile.ci_cpu b/docker/Dockerfile.ci_cpu
index e852b417ff..8e31b310fe 100644
--- a/docker/Dockerfile.ci_cpu
+++ b/docker/Dockerfile.ci_cpu
@@ -59,16 +59,9 @@ RUN bash /install/ubuntu_install_llvm.sh
COPY install/ubuntu_install_dnnl.sh /install/ubuntu_install_dnnl.sh
RUN bash /install/ubuntu_install_dnnl.sh
-COPY install/ubuntu_install_papi.sh /install/ubuntu_install_papi.sh
-RUN bash /install/ubuntu_install_papi.sh ""
-
-# wasmtime
-COPY install/ubuntu_install_wasmtime.sh /install/ubuntu_install_wasmtime.sh
-RUN bash /install/ubuntu_install_wasmtime.sh
-
-# AutoTVM deps
-COPY install/ubuntu_install_redis.sh /install/ubuntu_install_redis.sh
-RUN bash /install/ubuntu_install_redis.sh
+# MetaScheduler deps
+COPY install/ubuntu_install_xgboost.sh /install/ubuntu_install_xgboost.sh
+RUN bash /install/ubuntu_install_xgboost.sh
# BYODT deps
COPY install/ubuntu_install_universal.sh /install/ubuntu_install_universal.sh
@@ -86,19 +79,11 @@ RUN bash /install/ubuntu_install_tflite.sh
COPY install/ubuntu_install_jax.sh /install/ubuntu_install_jax.sh
RUN bash /install/ubuntu_install_jax.sh "cpu"
-# Compute Library
-COPY install/ubuntu_download_arm_compute_lib_binaries.sh
/install/ubuntu_download_arm_compute_lib_binaries.sh
-RUN bash /install/ubuntu_download_arm_compute_lib_binaries.sh
-
# sccache
COPY install/ubuntu_install_sccache.sh /install/ubuntu_install_sccache.sh
RUN bash /install/ubuntu_install_sccache.sh
ENV PATH=/opt/sccache:$PATH
-# Libxsmm deps
-COPY install/ubuntu_install_libxsmm.sh /install
-RUN bash /install/ubuntu_install_libxsmm.sh
-
# ONNX and PyTorch
COPY install/ubuntu_install_onnx.sh /install/ubuntu_install_onnx.sh
RUN bash /install/ubuntu_install_onnx.sh
diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu
index 2adb2cf5bf..df15215b94 100644
--- a/docker/Dockerfile.ci_gpu
+++ b/docker/Dockerfile.ci_gpu
@@ -94,6 +94,7 @@ RUN bash /install/ubuntu_install_tensorflow.sh
COPY install/ubuntu_install_jax.sh /install/ubuntu_install_jax.sh
RUN bash /install/ubuntu_install_jax.sh "cuda"
+# ONNX and PyTorch
COPY install/ubuntu_install_onnx.sh /install/ubuntu_install_onnx.sh
RUN bash /install/ubuntu_install_onnx.sh "cuda"
@@ -110,22 +111,14 @@ ENV NVIDIA_DRIVER_CAPABILITIES=compute,graphics,utility
COPY install/ubuntu_install_vulkan.sh /install/ubuntu_install_vulkan.sh
RUN bash /install/ubuntu_install_vulkan.sh
-# wasmtime
-COPY install/ubuntu_install_wasmtime.sh /install/ubuntu_install_wasmtime.sh
-RUN bash /install/ubuntu_install_wasmtime.sh
-
-# AutoTVM deps
-COPY install/ubuntu_install_redis.sh /install/ubuntu_install_redis.sh
-RUN bash /install/ubuntu_install_redis.sh
+# MetaScheduler deps
+COPY install/ubuntu_install_xgboost.sh /install/ubuntu_install_xgboost.sh
+RUN bash /install/ubuntu_install_xgboost.sh
# BYODT deps
COPY install/ubuntu_install_universal.sh /install/ubuntu_install_universal.sh
RUN bash /install/ubuntu_install_universal.sh
-# PAPI profiling deps
-COPY install/ubuntu_install_papi.sh /install/ubuntu_install_papi.sh
-RUN bash /install/ubuntu_install_papi.sh "cuda rocm"
-
# sccache
COPY install/ubuntu_install_sccache.sh /install/ubuntu_install_sccache.sh
RUN bash /install/ubuntu_install_sccache.sh
diff --git a/docker/install/ubuntu_download_arm_compute_lib_binaries.sh
b/docker/install/ubuntu_download_arm_compute_lib_binaries.sh
deleted file mode 100755
index d32a21c1fa..0000000000
--- a/docker/install/ubuntu_download_arm_compute_lib_binaries.sh
+++ /dev/null
@@ -1,66 +0,0 @@
-#!/bin/bash
-# Licensed to the Apache Software Foundation (ASF) under one
-# or more contributor license agreements. See the NOTICE file
-# distributed with this work for additional information
-# regarding copyright ownership. The ASF licenses this file
-# to you under the Apache License, Version 2.0 (the
-# "License"); you may not use this file except in compliance
-# with the License. You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied. See the License for the
-# specific language governing permissions and limitations
-# under the License.
-
-set -e
-architecture_type=$(uname -i)
-# Install cross-compiler when not building natively.
-# Depending on the architecture selected to compile for,
-# you may need to install an alternative cross-compiler.
-if [ "$architecture_type" != "aarch64" ]; then
- apt-get update
- apt-install-and-clear -y --no-install-recommends \
- g++-aarch64-linux-gnu \
- gcc-aarch64-linux-gnu
-fi
-
-compute_lib_version="v23.08"
-compute_lib_variant="arm64-v8a-neon"
-compute_lib_full_name="arm_compute-${compute_lib_version}-bin-linux-${compute_lib_variant}"
-compute_lib_base_url="https://github.com/ARM-software/ComputeLibrary/releases/download/${compute_lib_version}";
-compute_lib_file_name="${compute_lib_full_name}.tar.gz"
-compute_lib_download_url="${compute_lib_base_url}/${compute_lib_file_name}"
-
-target_lib="${compute_lib_variant}"
-
-# uncomment line below if you need asserts/debug version of the library
-# target_lib="${target_lib}-asserts"
-
-extract_dir="${compute_lib_full_name}"
-install_path="/opt/acl"
-
-tmpdir=$(mktemp -d)
-
-cleanup()
-{
- rm -rf "$tmpdir"
-}
-
-trap cleanup 0
-
-cd "$tmpdir"
-
-curl -sL "${compute_lib_download_url}" -o "${compute_lib_file_name}"
-tar xzf "${compute_lib_file_name}"
-
-rm -rf "${install_path}"
-mkdir -p "${install_path}"
-cp -r "${extract_dir}/include" "${install_path}/"
-cp -r "${extract_dir}/arm_compute" "${install_path}/include/"
-cp -r "${extract_dir}/support" "${install_path}/include/"
-cp -r "${extract_dir}/utils" "${install_path}/include/"
-cp -r "${extract_dir}/lib/${target_lib}" "${install_path}/lib"
diff --git a/docker/install/ubuntu_install_hexagon.sh
b/docker/install/ubuntu_install_hexagon.sh
deleted file mode 100755
index 57807398a7..0000000000
--- a/docker/install/ubuntu_install_hexagon.sh
+++ /dev/null
@@ -1,47 +0,0 @@
-#!/bin/bash
-# Licensed to the Apache Software Foundation (ASF) under one
-# or more contributor license agreements. See the NOTICE file
-# distributed with this work for additional information
-# regarding copyright ownership. The ASF licenses this file
-# to you under the Apache License, Version 2.0 (the
-# "License"); you may not use this file except in compliance
-# with the License. You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied. See the License for the
-# specific language governing permissions and limitations
-# under the License.
-
-set -o errexit -o nounset
-set -o pipefail
-
-# Install LLVM/clang
-CLANG_LLVM_HOME=/opt/clang-llvm
-LLVM_SHA=a9871772a8b13c1240a95a84a3327f84bb67dddc
-
-mkdir llvm-hexagon
-pushd llvm-hexagon
-git init
-git remote add origin https://github.com/llvm/llvm-project.git
-git fetch origin ${LLVM_SHA}
-git reset --hard FETCH_HEAD
-mkdir build
-pushd build
-cmake \
- -G Ninja \
- -DCMAKE_BUILD_TYPE=Release \
- -DCMAKE_INSTALL_PREFIX=${CLANG_LLVM_HOME} \
- -DLLVM_ENABLE_ASSERTIONS=ON \
- -DLLVM_TARGETS_TO_BUILD:STRING="Hexagon;X86" \
- -DLLVM_ENABLE_PROJECTS:STRING="llvm" \
- -DLLVM_DEFAULT_TARGET_TRIPLE=x86_64-unknown-linux-gnu \
- ../llvm
-ninja install
-
-popd
-popd
-rm -rf llvm-hexagon
diff --git a/docker/install/ubuntu_install_libxsmm.sh
b/docker/install/ubuntu_install_libxsmm.sh
deleted file mode 100755
index f0fda49248..0000000000
--- a/docker/install/ubuntu_install_libxsmm.sh
+++ /dev/null
@@ -1,32 +0,0 @@
-#!/bin/bash
-# Licensed to the Apache Software Foundation (ASF) under one
-# or more contributor license agreements. See the NOTICE file
-# distributed with this work for additional information
-# regarding copyright ownership. The ASF licenses this file
-# to you under the Apache License, Version 2.0 (the
-# "License"); you may not use this file except in compliance
-# with the License. You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied. See the License for the
-# specific language governing permissions and limitations
-# under the License.
-
-set -e
-set -u
-set -o pipefail
-
-pushd /usr/local/
-wget -q https://github.com/libxsmm/libxsmm/archive/refs/tags/1.17.tar.gz
-tar -xzf 1.17.tar.gz
-pushd ./libxsmm-1.17/
-make STATIC=0 -j$(($(nproc) - 1))
-cp -L include/* /usr/local/include/
-cp -L lib/*so /usr/local/lib/
-popd
-rm -rf 1.17.tar.gz libxsmm-1.17
-popd
diff --git a/docker/install/ubuntu_install_papi.sh
b/docker/install/ubuntu_install_papi.sh
deleted file mode 100755
index 2907aee180..0000000000
--- a/docker/install/ubuntu_install_papi.sh
+++ /dev/null
@@ -1,35 +0,0 @@
-#!/bin/bash
-# Licensed to the Apache Software Foundation (ASF) under one
-# or more contributor license agreements. See the NOTICE file
-# distributed with this work for additional information
-# regarding copyright ownership. The ASF licenses this file
-# to you under the Apache License, Version 2.0 (the
-# "License"); you may not use this file except in compliance
-# with the License. You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied. See the License for the
-# specific language governing permissions and limitations
-# under the License.
-
-set -e
-set -u
-set -o pipefail
-
-apt-get update --fix-missing
-
-# deps
-apt-install-and-clear -y linux-tools-common linux-tools-generic kmod
-
-cd /
-# Pulling the latest version of this has broken the images before. Checkout
the tagged version below for now.
-git clone --branch papi-6-0-0-1-t https://github.com/icl-utk-edu/papi
-cd papi/src
-export PAPI_CUDA_ROOT=/usr/local/cuda
-export PAPI_ROCM_ROOT=/opt/rocm
-./configure --with-components="$1"
-make -j $(nproc) && make install
diff --git a/docker/install/ubuntu_install_redis.sh
b/docker/install/ubuntu_install_redis.sh
deleted file mode 100755
index 6b0e093cb6..0000000000
--- a/docker/install/ubuntu_install_redis.sh
+++ /dev/null
@@ -1,25 +0,0 @@
-#!/bin/bash
-# Licensed to the Apache Software Foundation (ASF) under one
-# or more contributor license agreements. See the NOTICE file
-# distributed with this work for additional information
-# regarding copyright ownership. The ASF licenses this file
-# to you under the Apache License, Version 2.0 (the
-# "License"); you may not use this file except in compliance
-# with the License. You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied. See the License for the
-# specific language governing permissions and limitations
-# under the License.
-
-set -e
-set -u
-set -o pipefail
-
-apt-get update && apt-install-and-clear -y redis-server
-uv pip install \
- xgboost==3.2.0
diff --git a/docker/install/ubuntu_install_wasmtime.sh
b/docker/install/ubuntu_install_wasmtime.sh
deleted file mode 100755
index db9bd0b1c3..0000000000
--- a/docker/install/ubuntu_install_wasmtime.sh
+++ /dev/null
@@ -1,53 +0,0 @@
-#!/bin/bash
-# Licensed to the Apache Software Foundation (ASF) under one
-# or more contributor license agreements. See the NOTICE file
-# distributed with this work for additional information
-# regarding copyright ownership. The ASF licenses this file
-# to you under the Apache License, Version 2.0 (the
-# "License"); you may not use this file except in compliance
-# with the License. You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied. See the License for the
-# specific language governing permissions and limitations
-# under the License.
-
-set -euxo pipefail
-
-WASMTIME_VERSION="v41.0.3"
-WASMTIME_HOME=/opt/wasmtime
-
-case "$(uname -m)" in
- x86_64)
- WASMTIME_ARCH="x86_64-linux"
-
WASMTIME_SHA256="797d0a4f790e79c33ccaf43bfe413f077fff951e3a35145afe7b5a8324f14644"
- ;;
- aarch64|arm64)
- WASMTIME_ARCH="aarch64-linux"
-
WASMTIME_SHA256="1dd1f69089eeefc3826f38463f8375d6ff2e59684a2a85b44a6622516d0a5677"
- ;;
- *)
- echo "Unsupported architecture: $(uname -m)"
- exit 1
- ;;
-esac
-
-WASMTIME_TARBALL="wasmtime-${WASMTIME_VERSION}-${WASMTIME_ARCH}.tar.xz"
-WASMTIME_URL="https://github.com/bytecodealliance/wasmtime/releases/download/${WASMTIME_VERSION}/${WASMTIME_TARBALL}";
-
-TMP_DIR="$(mktemp -d)"
-cleanup() {
- rm -rf "${TMP_DIR}"
-}
-trap cleanup 0
-
-rm -rf "${WASMTIME_HOME}"
-mkdir -p "${WASMTIME_HOME}"
-download-and-verify "${WASMTIME_URL}" "${TMP_DIR}/${WASMTIME_TARBALL}" sha256
"${WASMTIME_SHA256}"
-
-tar -xJf "${TMP_DIR}/${WASMTIME_TARBALL}" -C "${WASMTIME_HOME}"
--strip-components=1
-ln -sf "${WASMTIME_HOME}/wasmtime" /usr/local/bin/wasmtime
diff --git a/docker/install/ubuntu_install_gluoncv.sh
b/docker/install/ubuntu_install_xgboost.sh
similarity index 95%
rename from docker/install/ubuntu_install_gluoncv.sh
rename to docker/install/ubuntu_install_xgboost.sh
index 434480257b..6cee4be8c9 100755
--- a/docker/install/ubuntu_install_gluoncv.sh
+++ b/docker/install/ubuntu_install_xgboost.sh
@@ -20,4 +20,4 @@ set -e
set -u
set -o pipefail
-uv pip install gluoncv==0.10.5.post0
+uv pip install xgboost==3.2.0
