hipudding be48528b06 CANN: Add broadcast for softmax and FA (#15208) ...

* refactor softmax

* fix fa

* fix mask shape

* format

* add comments

* Remove whitespace

2025-08-11 22:50:31 +08:00

..

acl_tensor.cpp

CANN: Implement GLU ops (#14884)

2025-07-26 17:56:18 +08:00

acl_tensor.h

CANN: Add the basic supports of Flash Attention kernel (#13627)

2025-05-26 10:20:18 +08:00

aclnn_ops.cpp

CANN: Add broadcast for softmax and FA (#15208)

2025-08-11 22:50:31 +08:00

aclnn_ops.h

CANN: Add ggml_set_rows (#14943)

2025-07-29 22:36:43 +08:00

CMakeLists.txt

CANN: add support for ACL Graph (#15065)

2025-08-06 14:12:42 +08:00

common.h

CANN: add support for ACL Graph (#15065)

2025-08-06 14:12:42 +08:00

Doxyfile

CANN: Add the basic supports of Flash Attention kernel (#13627)

2025-05-26 10:20:18 +08:00

ggml-cann.cpp

CANN: Add broadcast for softmax and FA (#15208)

2025-08-11 22:50:31 +08:00