freeliuzc ceafd757f0 [Speculative Decoding]Support multi-step mtp with cudagraph (#5624) (#5670) ...

* support multi-step mtp with cudagraph

* fix usage

* fix unit test

2025-12-23 13:18:47 +08:00

..

__init__.py

[LLM] First commit the llm deployment code

2025-06-09 19:20:15 +08:00

dcu_model_runner.py

[FDConfig]Remove splitwise_role and engine_worker_queue_port in FDConfig (#4147)

2025-09-19 17:01:52 +08:00

dcu_worker.py

[FDConfig]Remove total_block_num/dtype/block_size/enc_dec_block_num in ParallelConfig (#4400)

2025-10-16 20:00:37 +08:00

eplb.py

polish code with new pre-commit rule (#2923)

2025-07-19 23:19:27 +08:00

experts_manager.py

fix typos (#3684)

2025-09-01 17:50:17 +08:00

gcu_model_runner.py

[Models] Add forward_meta to moe models' forward function (#5138)

2025-12-04 13:26:58 +08:00

gcu_worker.py

[FDConfig]Remove total_block_num/dtype/block_size/enc_dec_block_num in ParallelConfig (#4400)

2025-10-16 20:00:37 +08:00

gpu_model_runner.py

[Speculative Decoding]Support multi-step mtp with cudagraph (#5624) (#5670)

2025-12-23 13:18:47 +08:00

gpu_worker.py

[FDConfig]Remove total_block_num/dtype/block_size/enc_dec_block_num in ParallelConfig (#4400)

2025-10-16 20:00:37 +08:00

hpu_model_runner.py

[Intel HPU] fix bug about RP 5138 (#5380)

2025-12-05 11:33:29 +08:00

hpu_worker.py

[FDConfig]Remove total_block_num/dtype/block_size/enc_dec_block_num in ParallelConfig (#4400)

2025-10-16 20:00:37 +08:00

iluvatar_model_runner.py

[Iluvatar GPU] fix ci error caused by rebuild_padding param and cuda graph (#4504)

2025-10-21 21:41:41 +08:00

iluvatar_worker.py

[Iluvatar GPU] Adapt VL model (#4313)

2025-10-17 16:13:38 +08:00

metax_model_runner.py

[Models] Add forward_meta to moe models' forward function (#5138)

2025-12-04 13:26:58 +08:00

metax_worker.py

[Metax] support ENABLE_V1_KVCACHE_SCHEDULER (#5163)

2025-11-24 19:19:49 +08:00

model_runner_base.py

[Feature] support chunked moe (#4575)

2025-12-01 15:17:18 +08:00

output.py

[Cherry-Pick] Support for request-level speculative decoding metrics monitoring.(#5518) (#5614)

2025-12-17 20:53:04 +08:00

tbo.py

[Feature] support Two batch overlap, mainly used in Prefill (#5078)

2025-12-05 14:58:50 +08:00

worker_base.py

[FDConfig]Remove splitwise_role and engine_worker_queue_port in FDConfig (#4147)

2025-09-19 17:01:52 +08:00

worker_process.py

[BugFix] fix rl signal (#5678)

2025-12-22 00:31:24 -08:00

xpu_model_runner.py

[XPU]Set top_p=0.0 by default on XPU to optimize performance (#5688)

2025-12-23 11:00:53 +08:00

xpu_worker.py

[xpu] support mtp for xpu(mix) (#5274)

2025-12-01 11:03:14 +08:00