From feec1ac7f9975bb4e1f34087a36f689050288c06 Mon Sep 17 00:00:00 2001 From: Kangyan-Zhou Date: Tue, 28 Apr 2026 19:32:38 -0700 Subject: [PATCH] ci: clean up stale-CUDA mooncake variant in install_extra_deps (#23960) Co-authored-by: Claude Opus 4.7 (1M context) --- scripts/ci/cuda/ci_install_dependency.sh | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/scripts/ci/cuda/ci_install_dependency.sh b/scripts/ci/cuda/ci_install_dependency.sh index bd0f6b650..1a595fbb1 100755 --- a/scripts/ci/cuda/ci_install_dependency.sh +++ b/scripts/ci/cuda/ci_install_dependency.sh @@ -370,11 +370,22 @@ stabilize_flashinfer_jit_paths() { install_extra_deps() { if [ "$CU_MAJOR" = "13" ]; then MOONCAKE_PKG="mooncake-transfer-engine-cuda13==0.3.10.post2" + MOONCAKE_STALE_PKG="mooncake-transfer-engine" EXTRA_NVIDIA_SPECS="nvidia-cuda-nvrtc" else MOONCAKE_PKG="mooncake-transfer-engine==0.3.10.post2" + MOONCAKE_STALE_PKG="mooncake-transfer-engine-cuda13" EXTRA_NVIDIA_SPECS="nvidia-cuda-nvrtc-cu12" fi + # Both variants own the same mooncake/ package files and bin/ scripts + # (mooncake_master, etc.). Uninstalling the stale variant deletes shared + # files that the live variant's RECORD still references, so we force a + # reinstall to restore them — pip would otherwise see "already satisfied" + # and skip. + if pip show ${MOONCAKE_STALE_PKG} >/dev/null 2>&1; then + $PIP_UNINSTALL_CMD ${MOONCAKE_STALE_PKG} $PIP_UNINSTALL_SUFFIX || true + $PIP_CMD install ${MOONCAKE_PKG} --force-reinstall --no-deps $PIP_INSTALL_SUFFIX + fi $PIP_CMD install ${MOONCAKE_PKG} ${EXTRA_NVIDIA_SPECS} py-spy scipy huggingface_hub[hf_xet] pytest $PIP_INSTALL_SUFFIX if [ "$IS_BLACKWELL" != "1" ]; then