software-engineering/spark-vllm-docker
2
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

saving flashinfer and vllm commits in wheels directories

Browse Source
This commit is contained in:
Eugene Rakhmatulin
2026-03-05 14:41:25 -08:00
parent a749fcce87
commit 2d03bc138d
2 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
Dockerfile
View File

@@ -113,7 +113,9 @@ RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \
cd flashinfer-cubin && uv build --no-build-isolation --wheel . --out-dir=/workspace/wheels -v && \ cd flashinfer-cubin && uv build --no-build-isolation --wheel . --out-dir=/workspace/wheels -v && \
# flashinfer-jit-cache # flashinfer-jit-cache
cd ../flashinfer-jit-cache && \ cd ../flashinfer-jit-cache && \
uv build --no-build-isolation --wheel . --out-dir=/workspace/wheels -v uv build --no-build-isolation --wheel . --out-dir=/workspace/wheels -v && \
# dump git ref in the wheels dir
cd .. && git rev-parse HEAD > /workspace/wheels/.flashinfer-commit
# ========================================================= # =========================================================
# STAGE 3: FlashInfer Wheel Export # STAGE 3: FlashInfer Wheel Export
@@ -196,7 +198,9 @@ RUN curl -L https://patch-diff.githubusercontent.com/raw/vllm-project/vllm/pull/
# Final Compilation # Final Compilation
RUN --mount=type=cache,id=ccache,target=/root/.ccache \ RUN --mount=type=cache,id=ccache,target=/root/.ccache \
--mount=type=cache,id=uv-cache,target=/root/.cache/uv \ --mount=type=cache,id=uv-cache,target=/root/.cache/uv \
uv build --no-build-isolation --wheel . --out-dir=/workspace/wheels -v uv build --no-build-isolation --wheel . --out-dir=/workspace/wheels -v && \
# dump git ref in the wheels dir
git rev-parse HEAD > /workspace/wheels/.vllm-commit
# ========================================================= # =========================================================
# STAGE 5: vLLM Wheel Export # STAGE 5: vLLM Wheel Export

4
build-and-copy.sh
View File

@@ -138,6 +138,9 @@ for a in assets:
for f in "$WHEELS_DIR/${PREFIX}"*.whl; do for f in "$WHEELS_DIR/${PREFIX}"*.whl; do
[ -f "$f" ] && mv "$f" "$DL_BACKUP/" [ -f "$f" ] && mv "$f" "$DL_BACKUP/"
done done
for f in "$WHEELS_DIR/.${PREFIX}"*; do
[ -f "$f" ] && mv "$f" "$DL_BACKUP/"
done
local URL NAME TMP_WHL local URL NAME TMP_WHL
local DOWNLOADED=() local DOWNLOADED=()
@@ -154,6 +157,7 @@ for a in assets:
if compgen -G "$DL_BACKUP/${PREFIX}*.whl" > /dev/null 2>&1; then if compgen -G "$DL_BACKUP/${PREFIX}*.whl" > /dev/null 2>&1; then
echo "Restoring previous $PREFIX wheels..." echo "Restoring previous $PREFIX wheels..."
mv "$DL_BACKUP/${PREFIX}"*.whl "$WHEELS_DIR/" mv "$DL_BACKUP/${PREFIX}"*.whl "$WHEELS_DIR/"
mv "$DL_BACKUP/.${PREFIX}"* "$WHEELS_DIR/"
fi fi
rm -rf "$DL_BACKUP" rm -rf "$DL_BACKUP"
return 1 return 1