youkaichao da02cb4b27 [core] further polish memory profiling (#12126) ...

Signed-off-by: youkaichao <youkaichao@gmail.com>

2025-01-18 12:25:08 +08:00

..

__init__.py

Change the name to vLLM (#150)

2023-06-17 03:07:40 -07:00

cache_engine.py

[Model] PP support for Mamba-like models (#10992)

2024-12-10 21:53:37 -05:00

cpu_enc_dec_model_runner.py

[torch.compile] Hide KV cache behind torch.compile boundary (#11677)

2025-01-10 13:14:42 +08:00

cpu_model_runner.py

[Hardware][CPU] Multi-LoRA implementation for the CPU backend (#11100)

2025-01-12 13:01:52 +00:00

cpu_pooling_model_runner.py

[torch.compile] Hide KV cache behind torch.compile boundary (#11677)

2025-01-10 13:14:42 +08:00

cpu_worker.py

[Hardware][CPU] Multi-LoRA implementation for the CPU backend (#11100)

2025-01-12 13:01:52 +00:00

enc_dec_model_runner.py

[torch.compile] Hide KV cache behind torch.compile boundary (#11677)

2025-01-10 13:14:42 +08:00

hpu_model_runner.py

[HPU][Bugfix] set_forward_context and CI test execution (#12014)

2025-01-14 11:04:18 +08:00

hpu_worker.py

[core] platform agnostic executor via collective_rpc (#11256)

2025-01-15 13:45:21 +08:00

model_runner_base.py

[platforms] enable platform plugins (#11602)

2024-12-30 20:24:45 +08:00

model_runner.py

[Misc] Add multipstep chunked-prefill support for FlashInfer (#10467)

2025-01-15 12:47:49 +08:00

multi_step_model_runner.py

[Misc] Add multipstep chunked-prefill support for FlashInfer (#10467)

2025-01-15 12:47:49 +08:00

multi_step_tpu_worker.py

[TPU] Implement multi-step scheduling (#8489)

2024-09-14 16:58:31 -07:00

multi_step_worker.py

[2/N] executor pass the complete config to worker/modelrunner (#9938)

2024-11-02 07:35:05 -07:00

neuron_model_runner.py

[Kernel] unified_attention for Attention.forward (#11967)

2025-01-13 19:28:53 +08:00

neuron_worker.py

[core] platform agnostic executor via collective_rpc (#11256)

2025-01-15 13:45:21 +08:00

openvino_model_runner.py

[Kernel] unified_attention for Attention.forward (#11967)

2025-01-13 19:28:53 +08:00

openvino_worker.py

[core] platform agnostic executor via collective_rpc (#11256)

2025-01-15 13:45:21 +08:00

pooling_model_runner.py

[torch.compile] Hide KV cache behind torch.compile boundary (#11677)

2025-01-10 13:14:42 +08:00

tpu_model_runner.py

[Kernel] unified_attention for Attention.forward (#11967)

2025-01-13 19:28:53 +08:00

tpu_worker.py

[Kernel] unified_attention for Attention.forward (#11967)

2025-01-13 19:28:53 +08:00

utils.py

[Doc][2/N] Reorganize Models and Usage sections (#11755)

2025-01-06 21:40:31 +08:00

worker_base.py

[core] allow callable in collective_rpc (#12151)

2025-01-17 20:47:01 +08:00

worker.py

[core] further polish memory profiling (#12126)

2025-01-18 12:25:08 +08:00

xpu_model_runner.py

[Kernel] unified_attention for Attention.forward (#11967)

2025-01-13 19:28:53 +08:00

xpu_worker.py

[XPU] Make pp group initilized for pipeline-parallelism (#11648)

2025-01-07 11:09:58 +08:00