| pytest aclgraph 功能添加
Co-authored-by: Yuanfeng_HW<yuanfeng24@huawei.com>
Co-authored-by: chenglongyu<chenglongyu@huawei.com>
# message auto-generated for no-merge-commit merge:
!5433 merge graph into master
pytest aclgraph 功能添加
Created-by: Yuanfeng_HW
Commit-by: chenglongyu;Yuanfeng_HW
Merged-by: cann-robot
Description:
描述
需求:针对算子flash_attn,添加图模式功能,主要修改点:
1、pytest测试脚本中添加了图模式调用功能;
2、torch_npu中进行flash_attn算子的图模式适配
关联的Issue
flash_attn训推归一新接口算子
测试
文档更新
类型标签
🐛 Bug 修复
✨ 新特性
⚡ 性能优化
♻️ 重构
🧪 测试
📦 构建/CI
🔧 配置变更
📝 文档更新
⬆️ 依赖升级
🔒 安全修复
🧹 代码清理
❓ 其他,请描述:
See merge request: cann/ops-transformer!5433 | 15 天前 |