Fix CI: add flashinfer --download-cubin to install dependencies (#18887)

Co-authored-by: Liangsheng Yin <lsyincs@gmail.com>
2026-02-16 13:50:10 -08:00
parent 1b659bcb08
commit f9c3def7fe
2 changed files with 35 additions and 0 deletions
@@ -289,6 +289,9 @@ if [ "$FLASHINFER_INSTALLED" = false ]; then
    exit 1
 fi

+# Download flashinfer cubins if the local set is incomplete
+bash "${SCRIPT_DIR}/ci_download_flashinfer_cubin.sh"
+
 # Show current packages
 $PIP_CMD list
 python3 -c "import torch; print(torch.version.cuda)"