Taneem Ibrahimandmergify[bot] <37929162+mergify[bot]@users.noreply.github.com> ba2ae9f239 [MRV2] Run pooling post processing on non-final PP ranks (#56666) ...

Signed-off-by: Taneem Ibrahim <taneem.ibrahim@gmail.com>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>

2026-09-14 16:11:11 -07:00

..

cpu

[CPU] Ship triton-cpu wheel and fix several hardcoded pin_memory=True (#52092)

2026-08-13 14:40:53 +08:00

gpu

[Model Runner V2] Acceptance estimation for adaptive verification (#52228)

2026-09-14 15:58:48 -07:00

sentinel

[Feature] Add fault tolerance framework (simplified) for DP+EP external LB deployments (#44428)

2026-07-25 11:49:09 -04:00

__init__.py

…

block_table.py

[5/N][warmup][DSv4] Migrate NVIDIA CuTeDSL attention kernels (#53566)

2026-09-12 00:05:59 -04:00

cp_utils.py

[PCP][Spec Decode] Adds PCP support for single-module MTP and replicated DSpark. (#56107)

2026-09-10 22:07:53 -07:00

cpu_model_runner.py

[6/N][warmup][DSv4] Migrate sampling, and DFlash JIT kernels (#56323)

2026-09-12 17:24:24 -07:00

cpu_worker.py

[CI/Build][CPU] Fix flaky rust downloads, broken prune flag, and triton-cpu cache coupling (#56247)

2026-09-10 22:27:26 +08:00

dp_utils.py

[Elastic EP] Reuse CUDA graphs across reconfiguration (#54985)

2026-09-12 10:26:39 -04:00

ec_connector_model_runner_mixin.py

[EPD] Add ECMooncakeConnector for encoder cache over Mooncake TransferEngine (#41567)

2026-09-07 06:21:43 +00:00

encoder_cudagraph_defs.py

[MM][CG] Enable encoder CUDA Graph for MiniCPM-V (#42785)

2026-09-10 06:28:48 +00:00

encoder_cudagraph.py

[MM][CG] Enable encoder CUDA Graph for MiniCPM-V (#42785)

2026-09-10 06:28:48 +00:00

gpu_input_batch.py

[Core][Model] Unify XD-RoPE into M-RoPE and derive the channel count (#56078)

2026-09-09 09:23:30 -07:00

gpu_model_runner.py

[6/N][warmup][DSv4] Migrate sampling, and DFlash JIT kernels (#56323)

2026-09-12 17:24:24 -07:00

gpu_ubatch_wrapper.py

[1/2][Model Runner V2] DBO support, eager mode (#50945)

2026-09-04 14:17:39 -07:00

gpu_worker.py

[MRV2] Run pooling post processing on non-final PP ranks (#56666)

2026-09-14 16:11:11 -07:00

kv_connector_model_runner_mixin.py

[3/N] HiSparse: host-resident sparse-MLA decode hot-buffering (#53781)

2026-09-12 00:36:47 -04:00

lora_model_runner_mixin.py

[Bugfix] Fix level-2 sleep/wake/reload with enable_lora=True (#39935)

2026-08-06 09:13:10 +08:00

mamba_utils.py

[K3 Bug] Fix Kimi-K3 RecoverSSM startup failure 'MambaAttentionBackendEnum.GDN_ATTN declares 4 states, but provides 2 state copy funcs' (#54634)

2026-08-31 23:22:49 +00:00

mm_encoder_model_runner.py

[Profiler] Add Proton CUDA graph attribution for MRV2 (#51084)

2026-09-14 19:02:07 +00:00

startup_plan.py

[Core] Persist and reuse the memory-profiling result across boots (opt-in) (#47388)

2026-07-08 00:53:02 +00:00

tpu_input_batch.py

fix: improve token_ids_cpu swap to copy only valid indices (#36255)

2026-08-28 17:22:39 -07:00

ubatch_utils.py

[Deprecation] Deprecate items scheduled for 0.29 (#55353)

2026-09-11 08:17:02 -07:00

ubatching.py

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

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

utils.py

[3/N] HiSparse: host-resident sparse-MLA decode hot-buffering (#53781)

2026-09-12 00:36:47 -04:00

worker_base.py

[Bugfix][DP] Synchronize the device on pause completion (#52914)

2026-08-26 09:50:33 -07:00

workspace.py

[Bugfix] Make DSV4 sparse MLA work end-to-end for plain decode, MTP, and DSpark (#51538)

2026-08-15 07:01:32 -07:00

xpu_model_runner.py

[XPU] Alias is_current_stream_capturing to XPU in cuda wrapper (#50526)

2026-08-05 15:10:17 +08:00

xpu_worker.py

[XPU] Fix device assignment for DP external LB (#53037)

2026-09-04 15:51:25 +08:00