Files

T

Michael GoinandCodex a7d00ec051 [Bugfix] DFlash fc sized wrong when num_target_layers != num_hidden_layers (#48524 )

Signed-off-by: mgoin <mgoin64@gmail.com>
Co-authored-by: Codex <noreply@openai.com>

2026-07-21 14:42:52 -07:00

metrics

[Model Runner V2] Support num NaNs in logits (#30187 )

2025-12-09 10:00:49 -08:00

[MRV2] Add encoder cache profiling implementation (#47985 )

2026-07-20 20:18:14 -07:00

model_states

[Bugfix] Fix DSA crash under breakable piecewise cudagraphs (#49302 )

2026-07-21 18:28:13 +01:00

pool

[Core] Move MRV1 late_interaction_runner.py out of MRV2 subtree (#48014 )

2026-07-08 18:30:11 +00:00

sample

Revert "[Sampler] Stop upcasting logits to fp32 in apply_sampling_params" (#48641 ) (#49033 )

2026-07-21 09:36:45 +01:00

spec_decode

[Bugfix] DFlash fc sized wrong when num_target_layers != num_hidden_layers (#48524 )

2026-07-21 14:42:52 -07:00

__init__.py

GPU Model Runner V2 (#25266 )

2025-11-21 08:20:55 -08:00

async_utils.py

[Perf] Use blocking CUDA events to avoid busy polling cuda driver lock (#47081 )

2026-07-07 09:36:10 +00:00

attn_utils.py

[3/N][KV-Cache Layout Refactor] Standardize Mamba cache; drop get_transfer_cache_regions (#44456 )

2026-07-21 09:16:15 +00:00

block_table.py

[Core] Add MRV2 virtual-batch PCP for MLA (#46570 )

2026-07-19 02:53:15 +00:00

buffer_utils.py

[Core] Ensure memory is pinned prior to async h2d copy (#45424 )

2026-06-20 20:02:24 -07:00

cp_utils.py

[Model Runner V2] Minor simplification for DCP (#34786 )

2026-02-18 11:04:53 -08:00

cudagraph_utils.py

[Bugfix] Fix DSA crash under breakable piecewise cudagraphs (#49302 )

2026-07-21 18:28:13 +01:00

dp_utils.py

[Model Runner V2][Bugfix] Fix MRV2 LoRA warmup (#35536 )

2026-06-15 13:17:23 -07:00

eplb_utils.py

[EPLB] Mask padding in EPLB load recording (#38128 )

2026-06-28 19:43:58 -07:00

input_batch.py

[Bugfix] Fix PD disagg + MTP correctness for Qwen3.5(GDN) (#47466 )

2026-07-07 13:51:22 +00:00

kv_connector.py

[Core] Cleanup KVConnector handling with PP + fix MRV2 (#43732 )

2026-05-28 13:12:03 -07:00

lora_utils.py

[Model Runner V2][Bugfix] Fix MRV2 LoRA warmup (#35536 )

2026-06-15 13:17:23 -07:00

model_runner.py

[MRV2] Add encoder cache profiling implementation (#47985 )

2026-07-20 20:18:14 -07:00

pcp_manager.py

[Core] Add MRV2 virtual-batch PCP for MLA (#46570 )

2026-07-19 02:53:15 +00:00

pp_utils.py

[XPU] Fix PP accuracy on XPU device (#47253 )

2026-07-07 09:17:27 +08:00

README.md

GPU Model Runner V2 (#25266 )

2025-11-21 08:20:55 -08:00

shutdown.py

[MRV2] Add shutdown() method (#41297 )

2026-05-02 21:06:30 -07:00

states.py

[Bugfix] Fix PD disagg + MTP correctness for Qwen3.5(GDN) (#47466 )

2026-07-07 13:51:22 +00:00

structured_outputs.py

[ModelRunner V2] Misc minor simplifications and optimizations (#33467 )

2026-02-01 22:17:14 +00:00

warmup.py

[ModelRunner V2] Warmup cross-attn properly in encoder-decoder case (#47308 )

2026-07-01 12:36:48 -07:00

README.md

[Experimental] Model Runner V2

This directory contains the new model runner which is under active development. Ping Woosuk Kwon for any changes.