GitLab

[MISC] Keep chunked prefill enabled by default with long context when prefix caching is enabled (#8342)

Co-authored-by: Michael Goin <michael@neuralmagic.com>

Co-authored-by: Harsha Bikki <harbikh@amazon.com>

Co-authored-by: Alexander Matveev <alexm@neuralmagic.com>

Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com>

Signed-off-by: Travis Johnson <tsjohnso@us.ibm.com>

Co-authored-by: afeldman-nm <156691304+afeldman-nm@users.noreply.github.com>

[Core] Fix tracking of model forward time to the span traces in case of PP>1 (#7440)

Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com>
Co-authored-by: Cyrus Leung <cyrus.tl.leung@gmail.com>

Signed-off-by: Wallas Santos <wallashss@ibm.com>
Co-authored-by: youkaichao <youkaichao@gmail.com>
Co-authored-by: Nick Hill <nickhill@us.ibm.com>
Co-authored-by: youkaichao <youkaichao@126.com>

[Core] Subclass ModelRunner to support cross-attention & encoder sequences (towards eventual encoder/decoder model support) (#4942)
Co-authored-by: Andrew Feldman <afeld2012@gmail.com>
Co-authored-by: Nick Hill <nickhill@us.ibm.com>

Co-authored-by: Michael Goin <michael@neuralmagic.com>

Co-authored-by: Michael Goin <michael@neuralmagic.com>

Co-authored-by: Roger Wang <ywang@roblox.com>

[Spec Decode] Disable Log Prob serialization to CPU for spec decoding for both draft and target models. (#6485)

Co-authored-by: Michael Goin <michael@neuralmagic.com>

Co-authored-by: Swapnil Parekh <swapnilp@ibm.com>
Co-authored-by: Joe G <joseph.granados@h2o.ai>
Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>

Signed-off-by: Xiaowei Jiang <xwjiang2010@gmail.com>
Co-authored-by: Roger Wang <ywang@roblox.com>

Signed-off-by: Xiaowei Jiang <xwjiang2010@gmail.com>
Co-authored-by: Cyrus Leung <cyrus.tl.leung@gmail.com>
Co-authored-by: Roger Wang <ywang@roblox.com>