Files

T

Giancarlo Delfin 5daf62271d [Model Runner V2] Fuse probabilistic rejection sample kernels (#38496 )

Signed-off-by: Giancarlo Delfin <gdelfin@inferact.ai>

2026-04-07 17:37:37 -07:00

metrics

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

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

[Core] Simplify multimodal masking (#34246 )

2026-04-01 01:18:22 -07:00

model_states

[Model Runner V2] Support Streaming Inputs (#37028 )

2026-03-20 20:42:25 +00:00

pool

[Frontend] Re-enable running MaxSim on GPU (#38620 )

2026-04-03 00:03:13 +08:00

sample

[Model Runner V2] Fuse probabilistic rejection sample kernels (#38496 )

2026-04-07 17:37:37 -07:00

spec_decode

[Model Runner V2] Fuse probabilistic rejection sample kernels (#38496 )

2026-04-07 17:37:37 -07:00

__init__.py

GPU Model Runner V2 (#25266 )

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

async_utils.py

[BugFix][MRV2] Fix cuda event reuse race (#39115 )

2026-04-07 00:21:09 +00:00

attn_utils.py

[MRV2] Fix for DS v3.2 (#38030 )

2026-03-24 14:03:24 -07:00

block_table.py

[MRV2] Avoid recompilation of _gather_block_tables_kernel (#37645 )

2026-03-20 10:31:45 -07:00

buffer_utils.py

[V1] Remove pin_memory() in async_copy_to_gpu to fix sporadic stalls (#37006 )

2026-03-14 01:37:32 +00: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][Minor] Simplify PP logic (#38031 )

2026-03-24 13:57:17 -07:00

dp_utils.py

[MRV2] Extensible CG dispatch rework (#35959 )

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

eplb_utils.py

[Feature] EPLB Support for GPU Model Runner v2 (#37488 )

2026-03-25 08:16:39 -07:00

input_batch.py

[Model Runner V2] Do not initialize sampler for non-last PP ranks (#36824 )

2026-03-12 03:55:28 +00:00

kv_connector.py

[MRV2][KVConnector] Fix missing build_connector_worker_meta (#38698 )

2026-04-03 08:42:52 +03:00

lora_utils.py

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

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

model_runner.py

[BugFix][MRV2] Fix cuda event reuse race (#39115 )

2026-04-07 00:21:09 +00: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] Support Streaming Inputs (#37028 )

2026-03-20 20:42:25 +00:00

structured_outputs.py

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

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

warmup.py

[MRV2] Consider spec decoding in warmup (#37812 )

2026-03-23 17:45:43 +00: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.