SunLei 3098aee05f [Perf] Support tensor transmission between work and engine with zero-copy to improve efficiency (#4839) ...

* feat(zmq): support tensor transmission with zero-copy for improved efficiency

* perf: zmq.send disable copy

* zmq recv data for debug

* convert logprobs tensor to cpu

2025-11-11 15:43:11 +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

remove input_ids from ForwardMeta (#4793)

2025-11-05 11:55:51 +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

[Perf] Support tensor transmission between work and engine with zero-copy to improve efficiency (#4839)

2025-11-11 15:43:11 +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

Optimizing the performance of think length limit using custom operators (#4279)

2025-10-20 21:09:13 +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

[Metax] support ERNIE-4.5-VL-28B (#4820)

2025-11-07 04:55:49 -08:00

metax_worker.py

[Metax] adapt DeepSeek (#4498)

2025-10-24 10:14:53 +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

[Perf] Support tensor transmission between work and engine with zero-copy to improve efficiency (#4839)

2025-11-11 15:43:11 +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

[TSP] Support qwen3 moe tsp + cudagraph (#4871)

2025-11-10 23:37:51 +08:00

xpu_model_runner.py

remove input_ids from ForwardMeta (#4793)

2025-11-05 11:55:51 +08:00

xpu_worker.py

support ep+tp at op layer (#4688)

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