cmcamdy 5a67a6d960 [XPU] support kernel for mtp(base) (#4748) ...

* [XPU] support kernel for mtp(base)

* [XPU] support kernel for mtp(base)

* format

* format

* format

* fix gather next token

* fix step && add test

* fix

* mv pre/post process

* add adjust batch / gather next token for mtp

* fix code style

* fix mtp kenrel name

* fix mtp kernel test

* mv xpu pre/post process

* mv xpu pre/post process

2025-11-27 15:05:44 +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

[Optimization] xgrammar async compile, multi thread, speed up (#4835)

2025-11-14 18:05:26 +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

[BugFix] fix mtp logprob bugs in chunk prefill (#5244)

2025-11-27 11:31:29 +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 bugs caused by other commits (#5074)

2025-11-17 15:28:55 +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

[Optimization] xgrammar async compile, multi thread, speed up (#4835)

2025-11-14 18:05:26 +08:00

metax_worker.py

[Metax] support ENABLE_V1_KVCACHE_SCHEDULER (#5163)

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

model_runner_base.py

[FDConfig]Remove max_num_batched_tokens/max_num_seqs in parallel config (#4116)

2025-09-17 10:43:35 +08:00

output.py

check (#5237)

2025-11-26 17:07:26 +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

[Feature] enable guided decoding ENABLE_V1_KVCACHE_SCHEDULER = 1 (#5140)

2025-11-26 10:22:35 +08:00

xpu_model_runner.py

[XPU] support kernel for mtp(base) (#4748)

2025-11-27 15:05:44 +08:00

xpu_worker.py

support ep+tp at op layer (#4688)

2025-11-05 11:15:57 +08:00