From d9cd77419873b4765c7123d2ba003dc96219232d Mon Sep 17 00:00:00 2001 From: Aarushi Jain <142941703+aarushjain29@users.noreply.github.com> Date: Fri, 24 Jul 2026 23:14:37 -0500 Subject: [PATCH] [ROCm][CI] Force native compile caches onto local disk (#49763) Signed-off-by: aarushjain29 Co-authored-by: Andreas Karatzas --- .buildkite/scripts/hardware_ci/run-amd-test.sh | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/.buildkite/scripts/hardware_ci/run-amd-test.sh b/.buildkite/scripts/hardware_ci/run-amd-test.sh index de3f443ad5..0f6bb2564c 100755 --- a/.buildkite/scripts/hardware_ci/run-amd-test.sh +++ b/.buildkite/scripts/hardware_ci/run-amd-test.sh @@ -400,10 +400,10 @@ initialize_native_environment() { native_root="/tmp/vllm-native-${job_id}" TMPDIR="/tmp/vllm-${job_id_suffix}/tmp" VLLM_RPC_BASE_PATH="/tmp" - : "${TORCHINDUCTOR_CACHE_DIR:=${native_root}/cache/torchinductor}" - : "${TRITON_CACHE_DIR:=${native_root}/cache/triton}" - : "${VLLM_CACHE_ROOT:=${native_root}/cache/vllm}" - : "${XDG_CACHE_HOME:=${native_root}/cache/xdg}" + TORCHINDUCTOR_CACHE_DIR="${native_root}/cache/torchinductor" + TRITON_CACHE_DIR="${native_root}/cache/triton" + VLLM_CACHE_ROOT="${native_root}/cache/vllm" + XDG_CACHE_HOME="${native_root}/cache/xdg" : "${HF_HOME:=/home/buildkite-agent/huggingface}" : "${HF_HUB_DOWNLOAD_TIMEOUT:=300}" : "${HF_HUB_ETAG_TIMEOUT:=60}" @@ -419,6 +419,8 @@ initialize_native_environment() { "${XDG_CACHE_HOME}" \ "${HF_HOME}" || return 1 + echo "Native compile caches: VLLM_CACHE_ROOT=${VLLM_CACHE_ROOT} TORCHINDUCTOR_CACHE_DIR=${TORCHINDUCTOR_CACHE_DIR}" + if [[ "${VLLM_CI_REQUIRE_PERSISTENT_HF_CACHE:-0}" == "1" ]]; then if ! command -v findmnt >/dev/null 2>&1; then echo "findmnt is required to verify the native Hugging Face cache mount" >&2