GitLab

This PR enables the fused topk_softmax kernel used in moe layer for HIP

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

Signed-off-by: Ye Cao <caoye.cao@alibaba-inc.com>

Co-authored-by: xuhao <xuhao@cambricon.com>

Co-authored-by: Zhuohan Li <zhuohan123@gmail.com>

Co-authored-by: Breno Faria <breno.faria@intrafind.com>

[Core] Cross-attention KV caching and memory-management (towards eventual encoder/decoder model support) (#4837)

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

This PR adds Triton kernel configs for the MoE kernel for MI300X

Co-authored-by: Ruth Evans <ruthevans@Ruths-MacBook-Pro.local>

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

Co-authored-by: beagleski <yunanzhang@microsoft.com>
Co-authored-by: bapatra <bapatra@microsoft.com>
Co-authored-by: Barun Patra <codedecde@users.noreply.github.com>
Co-authored-by: Michael Goin <michael@neuralmagic.com>

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

Co-authored-by: Cody Yu <hao.yu.cody@gmail.com>