Yufeng HeandMatthew Bonanni b2198670b1 [Bugfix] V1: support tuple model outputs in ubatch wrapper (dbo + spec decode) (#40789) ...

Signed-off-by: Yufeng He <40085740+he-yufeng@users.noreply.github.com>
Co-authored-by: Matthew Bonanni <mbonanni@redhat.com>

2026-05-13 15:47:51 -07:00

..

gpu

[ModelRunner V2] Share identical MTP weights (#42538)

2026-05-13 18:57:04 +00:00

__init__.py

[V1] Implement vLLM V1 [1/N] (#9289)

2024-10-22 01:24:07 -07:00

block_table.py

[Bugfix] Align block table for TRTLLM MLA edge-case (#39324)

2026-05-06 11:17:02 -07:00

cp_utils.py

[Bug] Fix dcp error message (#40053)

2026-04-20 10:52:32 -04:00

cpu_model_runner.py

[BugFix][CPU] fix error on CPU runner shutdown (#41034)

2026-04-28 21:08:35 -07:00

cpu_worker.py

[CPU] Refactor CPU affinity and memory management (#39781)

2026-04-17 21:01:08 +08:00

dp_utils.py

[Perf][1/n] Eliminate various GPU<->CPU syncs (#41429)

2026-05-11 20:36:03 +00:00

ec_connector_model_runner_mixin.py

[Refactor] Remove dead code in kv connector and model runner (#38383)

2026-03-31 17:05:23 -04:00

encoder_cudagraph_defs.py

[MM][Perf][CG] Support ViT full CUDA graph for Qwen3-VL video inference (#38061)

2026-04-14 16:49:32 +08:00

encoder_cudagraph.py

[MM][CG] Optimize default max_frames_per_batch auto-infer for ViT CUDA graph video inference (#40445)

2026-04-21 14:47:53 +00:00

gpu_input_batch.py

[Bugfix] Fix token loss in PP mode which causes degraded accuracy (#41133)

2026-05-06 14:07:32 -04:00

gpu_model_runner.py

[Spec Decode] Support hybrid attention models in extract_hidden_states (#39949)

2026-05-13 10:45:53 -07:00

gpu_ubatch_wrapper.py

[Bugfix] V1: support tuple model outputs in ubatch wrapper (dbo + spec decode) (#40789)

2026-05-13 15:47:51 -07:00

gpu_worker.py

[feat] Add explicit /start_weight_update and /finish_weight_update APIs for weight transfer (#39212)

2026-05-08 18:03:33 -07:00

kv_connector_model_runner_mixin.py

[Refactor] Remove dead code in kv connector and model runner (#38383)

2026-03-31 17:05:23 -04:00

lora_model_runner_mixin.py

[Bugfix] LoRA for DeepSeek V3.2 (#35077)

2026-04-22 19:33:50 +08:00

mamba_utils.py

[V0 Deprecation] Refactor kv cache from list to element (#37487)

2026-03-23 20:10:11 -07:00

tpu_input_batch.py

[Cleanup] Remove unused ModelRunner V1 InputBatch.num_tokens field (#30218)

2025-12-18 09:17:00 -08:00

ubatch_utils.py

[BugFix] Preserve max_seq_len in ubatch metadata during CUDA graph capture (#40961)

2026-05-05 04:29:34 +00:00

ubatching.py

[Misc] Replace Optional[X] with X | None syntax (#33332)

2026-01-30 01:56:59 -08:00

utils.py

[torch.compile]: Disable Sequence Parallelism (SP) for piecewise compilation (#38373)

2026-04-26 17:44:42 +00:00

worker_base.py

Measure encoder compile time seperate from llm backbone (#39240)

2026-04-14 10:52:49 -04:00

workspace.py

[Bugfix] Fix workspace resize leaking reserved GPU memory (#39226)

2026-04-23 20:50:05 +00:00

xpu_model_runner.py

[Refactor] Remove dead code in kv connector and model runner (#38383)

2026-03-31 17:05:23 -04:00

xpu_worker.py

[XPU] update dp rank w/o env-var isolation (#39856)

2026-05-12 14:49:54 +08:00