Files

Woosuk Kwon 8ccbcda5c0 [Model Runner V2] Remove unused warmup_for_prefill method (#36762 )

Signed-off-by: Woosuk Kwon <woosuk@inferact.ai>

2026-03-11 08:02:44 -07:00

metrics

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

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

[Misc] Rename group_mm_kwargs_by_modality -> group_and_batch_mm_kwargs (#36158 )

2026-03-06 12:32:48 +08:00

model_states

[Model Runner V2] Use unpadded num_tokens for PW CUDA graph attn metadata (#36626 )

2026-03-10 09:30:56 -07:00

pool

[Perf] Compute maxsim in worker side, reducing redundant copies, 2.7% E2E throughput improvement (#36159 )

2026-03-09 20:55:58 -07:00

sample

[ModelRunnerV2] Rename sampler functions and variables for clarity (#35459 )

2026-03-02 19:48:56 -08:00

spec_decode

[MRV2] Extensible CG dispatch rework (#35959 )

2026-03-09 13:58:45 -07:00

__init__.py

GPU Model Runner V2 (#25266 )

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

async_utils.py

[Model Runner V2] Fix pooling (#36019 )

2026-03-04 10:53:17 -08:00

attn_utils.py

[Model Runner V2] Support attention group (#35036 )

2026-02-21 16:42:53 -08:00

block_table.py

[Model Runner V2] Fix _compute_slot_mappings_kernel for chunked prefill (#36580 )

2026-03-10 00:23:42 -07:00

buffer_utils.py

[Model Runner V2] Misc code simplification (#35941 )

2026-03-04 15:26:35 -08:00

cp_utils.py

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

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

cudagraph_utils.py

[Model Runner V2] Use unpadded num_tokens for PW CUDA graph attn metadata (#36626 )

2026-03-10 09:30:56 -07:00

dp_utils.py

[MRV2] Extensible CG dispatch rework (#35959 )

2026-03-09 13:58:45 -07:00

input_batch.py

[MRV2] Extensible CG dispatch rework (#35959 )

2026-03-09 13:58:45 -07:00

kv_connector.py

[Spec Decode] Defer clearing KV connector metadata for EAGLE3 speculative decode + prefill / decode disagg setup (#34529 )

2026-02-22 08:08:32 -08:00

lora_utils.py

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

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

model_runner.py

[Model Runner V2] Remove unused warmup_for_prefill method (#36762 )

2026-03-11 08:02:44 -07:00

pp_utils.py

[Model Runner V2] A bit more PP simplification (#34766 )

2026-02-17 21:39:07 -08:00

README.md

GPU Model Runner V2 (#25266 )

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

states.py

[Model Runner V2] Avoid prepare prefill kernel launch overhead (#34780 )

2026-02-18 00:49:21 -08:00

structured_outputs.py

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

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

warmup.py

[Model Runner V2] Fix warmup for pipeline parallel (#36280 )

2026-03-06 16:58:51 -08: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.