Commit 3e2c262

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 30cc11d commit 3e2c262Copy full SHA for 3e2c262

2 files changed

+213

-340

lines changed

+213

-340

lines changed

Comments

(0)