Pleaplusone
df0ec55162
Disaggregate prefill for kv cache register style ( #950 )
...
### What this PR does / why we need it?
This PR adopt `LLMDataDist` for kv cache register and `pull_blocks`
style disaggregate prefill implementation. The interface implementation
mainly follows the design of NIXL PR
https://github.com/vllm-project/vllm/pull/17751/files#diff-7eaad0b7dee0626bf29d10081b0f0c5e3ea15a4af97e7b182a4e0d35f8346953
.
This PR can be test with the following step:
- Generate the rank table for all machine.
- execute`toy_proxy.py` to launch the disaggregate prefill proxy server,
specify the prefill ip, port and the decode ip, port
- Run the prefill server and decode server.
- send the request to the disaggregate prefill proxy
### Does this PR introduce _any_ user-facing change?
### How was this patch tested?
- vLLM version: v0.9.2
- vLLM main:
8d0a01a5f2pleaplusone.gy@gmail.com >
Signed-off-by: machenglong <machenglong_yewu@cmss.chinamobile.com >
Signed-off-by: liziyu179 <3475441767@qq.com >
Signed-off-by: underfitc <hucong24@huawei.com >
Signed-off-by: zouyida2052 <zouyida@huawei.com >
Signed-off-by: liziyu <liziyu16@huawei.com >
Signed-off-by: underfituu <hzhucong@163.com >
Co-authored-by: machenglong <machenglong_yewu@cmss.chinamobile.com >
Co-authored-by: liziyu179 <3475441767@qq.com >
Co-authored-by: underfitc <hucong24@huawei.com >
Co-authored-by: zouyida2052 <zouyida@huawei.com >
Co-authored-by: liziyu <liziyu16@huawei.com >
Co-authored-by: underfituu <hzhucong@163.com >
2025-07-26 17:15:47 +08:00
Mengqing Cao
20767a043c
[CI/UT] Fix disaggregated prefill ci ( #1313 )
...
### What this PR does / why we need it?
Use eager mode to run disaggregated prefill ci
### Does this PR introduce _any_ user-facing change?
N/A
### How was this patch tested?
CI passed with new existing test.
---------
Signed-off-by: MengqingCao <cmq0113@163.com >
2025-06-24 17:11:00 +08:00
wangxiyuan
cb341c7bcd
[CI] Fix PD job ( #1129 )
...
Fix e2e test for Pd job
Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com >
2025-06-09 16:34:41 +08:00
Mengqing Cao
6eddbd2521
[CI/UT][PD Disaggreate] Initialize PD Disaggreate UT ( #889 )
...
Initialize PD Disaggreate UT
---------
Signed-off-by: MengqingCao <cmq0113@163.com >
2025-05-29 10:17:12 +08:00
