Temporarily added PR2897 to Dockerfile

This commit is contained in:
Eugene Rakhmatulin
2026-03-31 22:06:08 -07:00
parent 23fb7dcc20
commit 3a3ab98b3e

View File

@@ -132,6 +132,16 @@ RUN curl -fsL https://github.com/flashinfer-ai/flashinfer/pull/2913.diff -o pr29
fi \ fi \
&& rm pr2913.diff && rm pr2913.diff
# TEMPORARY patch for flashinfer autotune and other improvements (PR 2897)
RUN curl -fsL https://github.com/flashinfer-ai/flashinfer/pull/2897.diff -o pr2897.diff \
&& if git apply --reverse --check pr2897.diff 2>/dev/null; then \
echo "PR #2897 already applied, skipping."; \
else \
echo "Applying FI PR #2897..."; \
git apply -v pr2897.diff; \
fi \
&& rm pr2897.diff
# Apply patch to avoid re-downloading existing cubins # Apply patch to avoid re-downloading existing cubins
COPY flashinfer_cache.patch . COPY flashinfer_cache.patch .
RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \ RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \