Skip to content

fix: bump flashinfer to 0.6.9 to match the V4-Flash MXFP4 minimum#65

Open
gvr13n wants to merge 1 commit into
kvcache-ai:mainfrom
gvr13n:fix/flashinfer-min-version-v4
Open

fix: bump flashinfer to 0.6.9 to match the V4-Flash MXFP4 minimum#65
gvr13n wants to merge 1 commit into
kvcache-ai:mainfrom
gvr13n:fix/flashinfer-min-version-v4

Conversation

@gvr13n

@gvr13n gvr13n commented Jul 12, 2026

Copy link
Copy Markdown

Fresh installs cannot serve DeepSeek-V4-Flash MXFP4: mxfp4_deepseek.py requires flashinfer >= 0.6.9 (_MIN_FI_VERSIONmxfp8_quantize, trtllm_fp4_block_scale_routed_moe), but pyproject.toml pins flashinfer_python/flashinfer_cubin ==0.6.3.

The failure mode is nasty to diagnose: the quant plugin fails its import-time version check with only a warning, the MXFP4 MoE method never registers, and the server later dies with an unrelated-looking AssertionError: Hidden size mismatch in the generic triton fused_experts during cuda-graph capture. (Hit this twice on a working V4-Flash + kt-hybrid box after a routine reinstall silently downgraded flashinfer per the pin.)

This bumps the pins and the Dockerfile FLASHINFER_VERSION arg together, preserving the documented jit-cache alignment.

Possibly worth a follow-up: promote the import-time version warning to a hard error when a V4 MXFP4 checkpoint is actually being served, so the failure surfaces at the true cause.

mxfp4_deepseek.py requires flashinfer >= 0.6.9 (_MIN_FI_VERSION: mxfp8_quantize,
trtllm_fp4_block_scale_routed_moe), but pyproject pins flashinfer_python/cubin
==0.6.3, so a fresh install cannot serve DeepSeek-V4-Flash MXFP4: the quant
plugin fails its import-time version check with a warning, the MXFP4 MoE method
never registers, and the server later crashes with an unrelated-looking
'AssertionError: Hidden size mismatch' in the generic triton fused_experts
during cuda-graph capture.

Bump the pins and the Dockerfile FLASHINFER_VERSION arg together, keeping the
documented jit-cache alignment.

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request updates the FLASHINFER_VERSION from 0.6.3 to 0.6.9 in the Dockerfile, and correspondingly bumps the flashinfer_python and flashinfer_cubin dependency versions to 0.6.9 in pyproject.toml. There are no review comments, and I have no feedback to provide.

Important

The consumer version of Gemini Code Assist on GitHub is being sunset. Starting June 18, 2026, new organization installations will be blocked, and all code review activity will officially cease on July 17, 2026.
For more details on the timeline and next steps, please review the Help Documentation.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant