qqtang/sskj
1
3
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Fix] Preserve FlashInfer wheel filename in image build

Browse Source
This commit is contained in:
Zhiyi Hong 2026-08-18 12:39:30 +08:00
parent daeffd147b
commit c14f8aa43a
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
experiments/pro6000/kimi3_pro6000_sglang_tp32ep32_moe_backend_prefill/Dockerfile
View File

@ -9,9 +9,10 @@ RUN rm -rf \
/usr/local/lib/python3.12/dist-packages/flashinfer \ /usr/local/lib/python3.12/dist-packages/flashinfer \
/usr/local/lib/python3.12/dist-packages/flashinfer_python-*.dist-info /usr/local/lib/python3.12/dist-packages/flashinfer_python-*.dist-info
COPY flashinfer_python-0.6.18-py3-none-any.whl /tmp/flashinfer_python.whl COPY flashinfer_python-0.6.18-py3-none-any.whl /tmp/flashinfer_python-0.6.18-py3-none-any.whl
RUN python3 -m pip install --no-deps --force-reinstall /tmp/flashinfer_python.whl && \ RUN python3 -m pip install --no-deps --force-reinstall \
rm -f /tmp/flashinfer_python.whl /tmp/flashinfer_python-0.6.18-py3-none-any.whl && \
rm -f /tmp/flashinfer_python-0.6.18-py3-none-any.whl
# The wheel contains Python/C++ sources. These three libraries are the exact # The wheel contains Python/C++ sources. These three libraries are the exact
# SM120 AOT artifacts that passed the Phase 3/4 correctness gate. # SM120 AOT artifacts that passed the Phase 3/4 correctness gate.
@ -31,4 +32,3 @@ RUN python3 -c "from pathlib import Path; p=Path('/sgl-workspace/sglang/python/s
LABEL ai.meta-stone.purpose="Kimi-K3 SM120 FlashInfer MXFP4 Phase 5 acceptance" \ LABEL ai.meta-stone.purpose="Kimi-K3 SM120 FlashInfer MXFP4 Phase 5 acceptance" \
ai.meta-stone.sglang.patch="fb929bb" \ ai.meta-stone.sglang.patch="fb929bb" \
ai.meta-stone.flashinfer.patch="7d1a4ee" ai.meta-stone.flashinfer.patch="7d1a4ee"