Skip to content

Commit 47f2c64

Browse files
committed
cann: update the docs CANN.md
1 parent c8c2908 commit 47f2c64

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

docs/backend/CANN.md

100644100755
Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -259,10 +259,6 @@ cmake --build build --config release
259259
Please add the **[CANN]** prefix/tag in issues/PRs titles to help the CANN-team check/address them without delay.
260260

261261
### Basic Flash Attention Support
262-
The basic FA kernel with aclnnops has been added in aclnn_ops.cpp.
263-
264-
Authors: Bizhao Shi ([email protected]), Yuxin Yang ([email protected]), Ruiyang Ma ([email protected]), and Guojie Luo ([email protected]).
265-
266-
267-
## TODO
268-
- Support more models and data types.
262+
The basic FA kernel with aclnnops has been added in aclnn_ops.cpp.
263+
Currently, the FA only supports the cases with FP16 KV tensors and NO logit softcap.
264+
Since the aclnn interface for flash attention cannot support the logit softcap, we will only update the

0 commit comments

Comments
 (0)