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)