From 28717e3d283d7f4eb9b2e912f5045da26db94392 Mon Sep 17 00:00:00 2001 From: YC Tseng Date: Mon, 9 Feb 2026 14:20:09 +0800 Subject: [PATCH] [AMD] CI - Fix AMD daily image release and install dependency (#18452) Co-authored-by: Bingxu Chen --- docker/rocm.Dockerfile | 5 +++-- scripts/ci/amd/amd_ci_install_dependency.sh | 4 +++- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/docker/rocm.Dockerfile b/docker/rocm.Dockerfile index d2e47479d..5124a2902 100644 --- a/docker/rocm.Dockerfile +++ b/docker/rocm.Dockerfile @@ -278,8 +278,9 @@ RUN /bin/bash -lc 'set -euo pipefail; \ printf "#!/usr/bin/env bash\nexec \"%s\" \"\$@\"\n" "$LLVM_CONFIG_PATH" > /usr/local/bin/llvm-config-16 && \ chmod +x /usr/local/bin/llvm-config-16; \ \ - # TVM Python bits need Cython + z3 before configure - "$VENV_PIP" install --no-cache-dir "cython>=0.29.36,<3.0" "apache-tvm-ffi>=0.1.6" "z3-solver>=4.13.0"; \ + # TVM Python bits need Cython + z3 before configure. + # Pin z3-solver==4.15.4.0: 4.15.4.0 has a manylinux wheel; 4.15.5.0 has no wheel and builds from source (fails: C++20 needs GCC 14+, image has GCC 11). + "$VENV_PIP" install --no-cache-dir "cython>=0.29.36,<3.0" "apache-tvm-ffi>=0.1.6" "z3-solver==4.15.4.0"; \ \ # Clone + pin TileLang (bundled TVM), then build git clone --recursive "${TILELANG_REPO}" /opt/tilelang && \ diff --git a/scripts/ci/amd/amd_ci_install_dependency.sh b/scripts/ci/amd/amd_ci_install_dependency.sh index bc3215b2e..b06c9638f 100755 --- a/scripts/ci/amd/amd_ci_install_dependency.sh +++ b/scripts/ci/amd/amd_ci_install_dependency.sh @@ -121,7 +121,9 @@ esac #docker exec -w / ci_sglang git clone https://github.com/merrymercy/human-eval.git git_clone_with_retry https://github.com/merrymercy/human-eval.git human-eval docker cp human-eval ci_sglang:/ -install_with_retry docker exec -w /human-eval ci_sglang pip install --cache-dir=/sgl-data/pip-cache -e . +# Ensure setuptools is installed (human-eval's setup.py imports pkg_resources) +docker exec ci_sglang pip install --cache-dir=/sgl-data/pip-cache setuptools +install_with_retry docker exec -w /human-eval ci_sglang pip install --cache-dir=/sgl-data/pip-cache --no-build-isolation -e . docker exec -w / ci_sglang mkdir -p /dummy-grok # Create dummy grok config inline (bypasses Azure blob storage which may have auth issues)