Files

T

df6c7ad031 [Core][Refactor] Move KV cache planning logic to kv_cache_planning.py (#61183 )

Signed-off-by: Dakai An <dakaian108@gmail.com>
Signed-off-by: Yifan Qiao <yifanqiao@inferact.ai>
Co-authored-by: Yifan Qiao <yifanqiao@inferact.ai>
Co-authored-by: Claude Opus 5.5 <noreply@anthropic.com>

2026-10-13 03:05:39 +00:00

metrics

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

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

[Perf][Multimodal] Sort multimodal encoder inputs by modality (#60196 )

2026-10-09 10:55:36 +08:00

model_states

[Bugfix][RecoverSSM] Correct aligned state indices at exact block boundaries (#59962 )

2026-10-12 02:08:14 +00:00

pool

[Bugfix][Pooling] Handle multimodal cache misses without crashing (#58975 )

2026-10-02 10:14:33 -04:00

sample

[Bugfix][MRV2] Use int64 index mappings to fix row-offset overflow (#60736 )

2026-10-09 14:38:11 -07:00

spec_decode

[Bug] Honor spec decode draft model config (#61240 )

2026-10-12 15:13:23 -04:00

__init__.py

GPU Model Runner V2 (#25266 )

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

async_utils.py

[BugFix][DiffusionGemma][CPU] Copy async output snapshots instead of aliasing reused buffers (#59829 )

2026-10-08 19:16:51 -07:00

attn_utils.py

[XPU] limit FA kernel block size to multipleOf 64 (#57297 )

2026-10-12 10:25:32 +08:00

block_table.py

[KV Cache] GLM-5.3-Flash: use the generic packed KV layout; CircularBufferSpec tail with speculative ring slots (#57169 )

2026-10-09 15:48:37 -04:00

buffer_utils.py

[Bugfix][MRV2] Use int64 index mappings to fix row-offset overflow (#60736 )

2026-10-09 14:38:11 -07:00

cp_utils.py

[MRV2] Miscellaneous code cleanup (#57980 )

2026-09-23 01:51:52 -07:00

cudagraph_utils.py

[Core][Refactor] Move KV cache planning logic to kv_cache_planning.py (#61183 )

2026-10-13 03:05:39 +00:00

dp_utils.py

[MRV2] Miscellaneous code structure updates (#60754 )

2026-10-09 17:58:09 -07:00

ec_connector.py

[MRV2] Miscellaneous code cleanup (#57980 )

2026-09-23 01:51:52 -07:00

eplb_utils.py

[EPLB] Add EPLB support to dspark draft models (DeepSeekV4) (#57549 )

2026-10-12 10:03:09 -04:00

input_batch.py

[Bugfix][MRV2] Use int64 index mappings to fix row-offset overflow (#60736 )

2026-10-09 14:38:11 -07:00

kv_connector.py

[MRV2] Miscellaneous code structure updates (#60754 )

2026-10-09 17:58:09 -07:00

lora_utils.py

Add pydocstyle to the ruff rules (#52136 )

2026-09-16 16:44:29 +00:00

model_runner.py

[Bugfix][MRV2] Fix FlashInfer attention and ReplaySSM warmups on MRV2 (#61013 )

2026-10-11 16:04:41 -07:00

pcp_manager.py

[Bugfix][MRV2] Use int64 index mappings to fix row-offset overflow (#60736 )

2026-10-09 14:38:11 -07:00

pp_utils.py

[Perf][PP] Skip sampled-token broadcasts whose requests leave the engine (#58542 )

2026-09-30 19:18:55 -07: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

[ModelRunner V2] Speculative Decoding NGram GPU Implementations (#40704 )

2026-10-06 13:32:59 -07:00

structured_outputs.py

[Bugfix] Don't let a structured-output request sample from an unmasked row (#54442 )

2026-10-01 02:11:44 +00:00

ubatch_utils.py

[Perf][PP] Skip sampled-token broadcasts whose requests leave the engine (#58542 )

2026-09-30 19:18:55 -07:00

warmup.py

[Attention] Remove cascade attention (#61126 )

2026-10-12 10:41:07 -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.