[ROCm][CI] Force native compile caches onto local disk (#49763)
Signed-off-by: aarushjain29 <Aarushi.Jain2@amd.com> Co-authored-by: Andreas Karatzas <akaratza@amd.com>
Aarushi Jain committed
Jul 24, 2026 at 23:14 UTC
d9cd77419873b4765c7123d2ba003dc96219232d
1 file changed
+6
-4
.buildkite/scripts/hardware_ci/run-amd-test.sh
+6
-4
@@ -400,10 +400,10 @@ initialize_native_environment() {
400
native_root="/tmp/vllm-native-${job_id}"
401
TMPDIR="/tmp/vllm-${job_id_suffix}/tmp"
402
VLLM_RPC_BASE_PATH="/tmp"
403
- : "${TORCHINDUCTOR_CACHE_DIR:=${native_root}/cache/torchinductor}"
404
- : "${TRITON_CACHE_DIR:=${native_root}/cache/triton}"
405
- : "${VLLM_CACHE_ROOT:=${native_root}/cache/vllm}"
406
- : "${XDG_CACHE_HOME:=${native_root}/cache/xdg}"
403
+ TORCHINDUCTOR_CACHE_DIR="${native_root}/cache/torchinductor"
404
+ TRITON_CACHE_DIR="${native_root}/cache/triton"
405
+ VLLM_CACHE_ROOT="${native_root}/cache/vllm"
406
+ XDG_CACHE_HOME="${native_root}/cache/xdg"
407
: "${HF_HOME:=/home/buildkite-agent/huggingface}"
408
: "${HF_HUB_DOWNLOAD_TIMEOUT:=300}"
409
: "${HF_HUB_ETAG_TIMEOUT:=60}"
@@ -419,6 +419,8 @@ initialize_native_environment() {
419
"${XDG_CACHE_HOME}" \
420
"${HF_HOME}" || return 1
421
422
+ echo "Native compile caches: VLLM_CACHE_ROOT=${VLLM_CACHE_ROOT} TORCHINDUCTOR_CACHE_DIR=${TORCHINDUCTOR_CACHE_DIR}"
423
+
424
if [[ "${VLLM_CI_REQUIRE_PERSISTENT_HF_CACHE:-0}" == "1" ]]; then
425
if ! command -v findmnt >/dev/null 2>&1; then
426
echo "findmnt is required to verify the native Hugging Face cache mount" >&2