Skip to content

follow-up: vLLM Kimi-K3 W4A8 MegaMoE (deep_gemm) data lane — unblocked on post-merge vllm main #1462

Description

@tianhaox

Context

Kimi-K3's checkpoint (compressed-tensors mxfp4-pack + SiTU) serves two vLLM MoE paths:

  1. Single-node / TP: MarlinExperts weight-only W4A16 (CUTLASS W4A4's activation whitelist has no SITU) — collected and shipped in feat: Kimi-K3 support — hybrid KDA+MLA model, DSPARK, multi-arch sglang + vllm perf data (8 systems) #1435 on all eight systems, kernel lane w4a16_mxfp4.
  2. Recommended DEP scale-out (--moe-backend deep_gemm_mega_moe): DeepGEMM MegaMoE, fp8 activations × fp4 weights = W4A8 semantics. This path was unrunnable on the pinned preview image (vllm/vllm-openai@sha256:e90e2603, 0.1.dev19262 — no deep_gemm package), so feat: Kimi-K3 support — hybrid KDA+MLA model, DSPARK, multi-arch sglang + vllm perf data (8 systems) #1435 carries no vLLM W4A8 rows; W4A8 scope is covered by the sglang trtllm-gen lane only.

What changed upstream (2026-08-02 research)

Follow-up work (the actual scope — large, hence this issue)

  • Pick and pin a post-merge image (nightly or the first release containing K3). A pin bump triggers the standing full kda dispatch re-audit (merged-main routing may differ from the preview branch: FlashKDA/fused_kda_decode availability per SM, backend selector changes).
  • vLLM MegaMoE module lane collector (peer of the sglang MegaMoE module lane): needs 8-GPU Blackwell + --enable-expert-parallel; verify KimiK3MegaMoEExperts construction against serving (SiTU, latent MoE, grouped top-k).
  • Collect W4A8 rows on Blackwell systems; decide version-dir/labeling for nightly builds (setuptools-scm dev versions).
  • Re-run the K3 vllm serving-truth audit on the new pin (Marlin fallback vs CUTLASS vs trtllm-gen routing for the compressed-tensors path).
  • Support-matrix / SDK: decide whether the vllm K3 resolver should query w4a8 on Blackwell once data exists (mirroring resolve_kimi_k3_moe_arch_mode for sglang).

References

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions