如题。目前我的项目需求是在 Atlas200I DK A2 开发板上进行 FlashAttention 的推理。但是 CANN 包内没有提供在 Ascend310B1 上编译部署好的 FlashAttention 算子 ,所以我想参考 Ascend C 的项目示例(https://www.hiascend.com/document/detail/zh/canncommercial/80RC1/developmentguide/opdevg/Ascendcopdevg/atlas_ascendc_10_0026.html),自己定义一个 FlashAttention 算子。想要参考一份基于 Ascend C 的 FlashAttention 实现,但是我在 github 和 gitee 上没有找到相关的开源实现,请问论坛里有大神有相关的开源资料参考吗?
请问有无开源版本的基于 Ascend C 的 FlashAttention 实现?
已解决



