FastDeploy/fastdeploy/model_executor/layers/sample/sampler.py at 8e0f4dfd0cf70a0a67aca59629089b4a193404bb

mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-12-24 13:28:13 +08:00

Files

cmcamdy 9f4977eb74 [xpu] support mtp for xpu(mix) (#5274 )

* [XPU] support kernel for mtp(base)

* [XPU] support kernel for mtp(base)

* format

* format

* format

* fix gather next token

* fix step && add test

* fix

* mv pre/post process

* add adjust batch / gather next token for mtp

* fix code style

* fix mtp kenrel name

* fix mtp kernel test

* mv xpu pre/post process

* mv xpu pre/post process

* [xpu] support mtp

* fix code style

2025-12-01 11:03:14 +08:00

45 KiB

Raw Blame History

View Raw

45 KiB Raw Blame History

45 KiB

Raw Blame History