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

MXFP4 fix cache bug

Browse Source
This commit is contained in:
Eugene Rakhmatulin
2026-02-18 16:53:57 -08:00
parent c60c16e867
commit b959818536
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
Dockerfile.mxfp4
View File

@@ -109,13 +109,13 @@ RUN --mount=type=cache,id=git-flashinfer,target=/git-cache/flashinfer \
echo "=== Using cached FlashInfer repo ===" && \ echo "=== Using cached FlashInfer repo ===" && \
cp -a /git-cache/flashinfer /workspace/flashinfer && \ cp -a /git-cache/flashinfer /workspace/flashinfer && \
cd /workspace/flashinfer && \ cd /workspace/flashinfer && \
git fetch origin; \ git fetch origin && git fetch origin --tags; \
else \ else \
echo "=== Cloning FlashInfer (first build) ===" && \ echo "=== Cloning FlashInfer (first build) ===" && \
git clone ${FLASHINFER_REPO} /workspace/flashinfer && \ git clone ${FLASHINFER_REPO} /workspace/flashinfer && \
cp -a /workspace/flashinfer /git-cache/flashinfer; \ cp -a /workspace/flashinfer /git-cache/flashinfer; \
fi && \ fi && \
cd /workspace/flashinfer && git checkout ${FLASHINFER_SHA} cd /workspace/flashinfer && (git checkout --detach origin/${FLASHINFER_SHA} 2>/dev/null || git checkout ${FLASHINFER_SHA})
# Clone spdlog submodule (small, no caching needed) # Clone spdlog submodule (small, no caching needed)
RUN cd /workspace/flashinfer && \ RUN cd /workspace/flashinfer && \
@@ -129,14 +129,14 @@ RUN --mount=type=cache,id=git-cutlass,target=/git-cache/cutlass \
echo "=== Using cached CUTLASS repo ===" && \ echo "=== Using cached CUTLASS repo ===" && \
cp -a /git-cache/cutlass 3rdparty/cutlass && \ cp -a /git-cache/cutlass 3rdparty/cutlass && \
cd 3rdparty/cutlass && \ cd 3rdparty/cutlass && \
git fetch origin; \ git fetch origin && git fetch origin --tags; \
else \ else \
echo "=== Cloning CUTLASS (first build) ===" && \ echo "=== Cloning CUTLASS (first build) ===" && \
rm -rf /git-cache/cutlass/* /git-cache/cutlass/.* 2>/dev/null || true && \ rm -rf /git-cache/cutlass/* /git-cache/cutlass/.* 2>/dev/null || true && \
git clone ${CUTLASS_REPO} 3rdparty/cutlass && \ git clone ${CUTLASS_REPO} 3rdparty/cutlass && \
cp -a /workspace/flashinfer/3rdparty/cutlass/. /git-cache/cutlass/; \ cp -a /workspace/flashinfer/3rdparty/cutlass/. /git-cache/cutlass/; \
fi && \ fi && \
cd /workspace/flashinfer/3rdparty/cutlass && git checkout ${CUTLASS_SHA} cd /workspace/flashinfer/3rdparty/cutlass && (git checkout --detach origin/${CUTLASS_SHA} 2>/dev/null || git checkout ${CUTLASS_SHA})
# Build FlashInfer wheels # Build FlashInfer wheels