diff --git a/scripts/build-gpu-image.sh b/scripts/build-gpu-image.sh index ebd08bcb8..299678584 100755 --- a/scripts/build-gpu-image.sh +++ b/scripts/build-gpu-image.sh @@ -391,16 +391,23 @@ if matches: print(matches[-1]) PY )" -prewarm_image="${pull_image_repo}:${image_tag}" +prewarm_tag_image="${pull_image_repo}:${image_tag}" +prewarm_image="${prewarm_tag_image}" +prewarm_refresh_tag_image="" if [[ -n "${image_digest}" ]]; then if [[ "${pull_image_repo}" == "${image_repo}" ]]; then prewarm_image="${pull_image_repo}@${image_digest}" + prewarm_refresh_tag_image="${prewarm_tag_image}" else echo "Prewarm pull repo differs from pushed image repo; using mutable tag for pull-through freshness:" echo " Pushed image digest: ${image_repo}@${image_digest}" echo " Prewarm image: ${prewarm_image}" fi fi +prewarm_display="${prewarm_image}" +if [[ -n "${prewarm_refresh_tag_image}" ]]; then + prewarm_display="${prewarm_image} and refreshing ${prewarm_refresh_tag_image}" +fi dump_prewarm_diagnostics() { echo "::group::Prewarm diagnostics" @@ -427,6 +434,31 @@ sanitize_k8s_name_part() { | cut -c1-35 } +render_prewarm_init_containers() { + cat </dev/null 2>&1 || true "${kubectl_cmd[@]}" apply -n "${prewarm_namespace}" -f - <