Commit 1ed5aa4

authored and

committed

CANN: Add broadcast for softmax and FA (llama/15208)

* refactor softmax * fix fa * fix mask shape * format * add comments * Remove whitespace

1 parent 3b3e6fc commit 1ed5aa4Copy full SHA for 1ed5aa4

2 files changed

+213

-340

lines changed

+213

-340

lines changed

Comments

(0)