[cherry-pick] Fix softmax cuda bug (#21720) (#22160)
* Fix softmax cuda bug * Refine multihead log and softmax logic * Align block to 32
Showing
想要评论请 注册 或 登录
* Fix softmax cuda bug * Refine multihead log and softmax logic * Align block to 32