GitLab

[Core][2/N] Model runner refactoring part 2. Combine prepare prefill / decode to a single API (#4681)

This PR combines prepare_prompt and prepare_decode into a single API. This PR also coelsce the attn metadata for prefill/decode to a single class and allow to slice them when running attn backend.

It also refactors subquery_start_loc which was not refactored in the previous PR

Co-authored-by: SAHIL SUNEJA <suneja@us.ibm.com>

[Frontend] [Core] perf: Automatically detect vLLM-tensorized model, update `tensorizer` to version 2.9.0 (#4208)

Co-authored-by: Robert Shaw <114415538+robertgshaw2-neuralmagic@users.noreply.github.com>

Co-authored-by: Cade Daniel <edacih@gmail.com>

Co-authored-by: Lei Wen <wenlei03@qiyi.com>

Co-authored-by: Lei Wen <wenlei03@qiyi.com>

Co-authored-by: Robert Shaw <114415538+robertgshaw2-neuralmagic@users.noreply.github.com>
Co-authored-by: Robert Shaw <rshaw@neuralmagic.com>

Co-authored-by: DefTruth <31974251+deftruth@users.noreply.github.com>

Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>

Co-authored-by: Danny Guinther <dguinther@neuralmagic.com>

Signed-off-by: Tao He <sighingnow@gmail.com>

Co-authored-by: Yun Ding <yunding@nvidia.com>
Co-authored-by: Roger Wang <ywang@roblox.com>

Co-authored-by: Harry Mellor <hmellor@oxts.com>