Release v1.4: Grid188 decode and runtime synchronization fixes - #2
Merged
Merged
Conversation
This was referenced Jul 17, 2026
lukealonso
added a commit
to local-inference-lab/b12x
that referenced
this pull request
Jul 18, 2026
Add a two-tier heterogeneous-quantization mode to the W4A16 fused MoE as trace-time specialization of the existing kernel instead of a parallel kernel path: - emit_tile hook at _run_persistent_gemm's expert-resolution seam and a schedule_whole_tiles constexpr mode; every existing config compiles to byte-identical objects (verified against a captured baseline matrix). - W4A16FusedMoeHybridKernel composes two per-tier W4A16FusedMoeKernels and dispatches per-route via a global-expert descriptor map (tier<<8|local), reusing _moe_body, the grid barrier, activation, and _run_tile unchanged. - compile_w4a16_fused_moe_hybrid with fail-closed spill admission, a graph-safe b12x::w4a16_fused_moe_hybrid_launch op, run_w4a16_moe_hybrid, and build_w4a16_tier_local_map. - _w4a16_fused_persistent_grid_x minimizes the FC1+FC2 whole-tile critical path (cap grid beats FC1-right-sizing: 95.9us vs 107.1us at m=4). GLM-5.2 TP4 shard geometry (64 NVFP4 + 192 NF3, H6144/I512/topk8), graph replays vs the serial two-launch path: 1.75x at m=4 (167.5 -> 95.9us), 1.98x at m=1. Parity suite covers mixed/single-tier/invalid routes, both schedules, partial batches; 122 regs/thread, zero local memory. One-grid schedule and exact-geometry admission concepts derived from the Grid188 port in PR #36 by voipmonitor (davidsyoung/vllm-glm52#2). Co-Authored-By: Claude Fable 5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01JLscs7Yzm5B6Gxfij72md5
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
How the optimizations work
The hybrid decode and post-warmup paths are enabled in source and in the image. Explicit
0/false environment overrides remain available for diagnosis; normal operation no longer depends on Compose-only gates.Benchmarks
Measured on 4× RTX PRO 6000 Blackwell 96 GB, TP4+DCP4, MTP3 probabilistic draft sampling:
Aggregate and per-user throughput are distinct metrics. Deterministic and stochastic payloads are reported separately, not compared as an A/B.
BENCHMARKS.mdcontains the run-level values and interpretation limits.Verification
py_compile.docker compose config -qpassed for the final Compose file.No new regression-test framework is included; the source-equivalent runtime stack is covered by the recorded full-model benchmark evidence, and the PR packaging was verified through the final image build.