[PD Disaggregation] Fix cache messager performance problem & add kv transfer benchmark tool #6434
+461
−4
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Motivation
src_block_idsto paddle.Tensor in Python caused significant performance degradation in pybind parsing. The issue was identified during KV cache transfer optimization work.Modifications
Performance Fix in
cache_messager.py:paddle.to_tensor()conversion forsrc_block_idsanddest_block_idstime.time()totime.perf_counter()for more accurate performance measurementAdded Benchmark Tool:
benchmark.pyfor comprehensive KV cache transfer performance testingUsage or Command
# Performance test after fix python benchmark.py --splitwise_role prefill --decode_ip 127.0.0.1 --num_layers 61 --blocks_per_transfer 64Accuracy Tests
Checklist
[FDConfig],[APIServer],[Engine],[Scheduler],[PD Disaggregation],[Executor],[Graph Optimization],[Speculative Decoding],[RL],[Models],[Quantization],[Loader],[OP],[KVCache],[DataProcessor],[BugFix],[Docs],[CI],[Optimization],[Feature],[Benchmark],[Others],[XPU],[HPU],[GCU],[DCU],[Iluvatar],[Metax]]pre-commitbefore commit.releasebranch, make sure the PR has been submitted to thedevelopbranch, then cherry-pick it to thereleasebranch with the[Cherry-Pick]PR tag.