Skip to content

Commit b11d639

Browse files
hipuddingggerganov
authored andcommitted
CANN: Add broadcast for softmax and FA (llama/15208)
* refactor softmax * fix fa * fix mask shape * format * add comments * Remove whitespace
1 parent f9070d6 commit b11d639

File tree

2 files changed

+213
-340
lines changed

2 files changed

+213
-340
lines changed

0 commit comments

Comments
 (0)