cann-robotfix: 修正 attention_update CheckShape_95 日志中错误变量名 lse → localOut

文件	最后提交记录	最后更新时间
docs	doc Tools扫描错误修改 Co-authored-by: gitee-yanglulu<yanglulul@h-partners.com> # message auto-generated for no-merge-commit merge: !3348 merge master into master doc Tools扫描错误修改 Created-by: gitee-yanglulu Commit-by: gitee-yanglulu Merged-by: cann-robot Description: doc Tools扫描错误修改 See merge request: cann/ops-transformer!3348	5 个月前
examples	匹配最新cann版本，上线新版本attention/ffn/moe/posembedding算子 Co-authored-by: huangchuhong<huangchuhong1@h-partners.com> # message auto-generated for no-merge-commit merge: !538 merge master into master 匹配最新cann版本，上线新版本attention/ffn/moe/posembedding算子 Created-by: huang-chuhong Commit-by: huangchuhong Merged-by: cann-robot Description: ## 描述 transformer 同步算子最新开发，涉及attention/ffn/moe/posembedding ## 关联的Issue https://gitcode.com/cann/ops-transformer/issues/315 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-transformer!538	8 个月前
op_graph	fix: add add_graph_plugin_sources() to enable proto generation for 6 operators Co-authored-by: cao-ren-lei<caorenlei@huawei.com> Co-authored-by: wangxun21<wangxun21@huawei.com> # message auto-generated for no-merge-commit merge: !5231 merge cherry-pick-4905-op_proto into master fix: add add_graph_plugin_sources() to enable proto generation for 6 operators Created-by: wangxun21 Commit-by: wangxun21;cao-ren-lei Merged-by: cann-robot Description: ## 描述为6个算子的 op_graph CMakeLists.txt 添加 add_graph_plugin_sources() 调用，启用 proto 文件生成。改动原因：上述6个算子的 op_graph 目录下缺少 add_graph_plugin_sources() 调用，导致编译时无法生成对应的 proto 文件。改动方法：在各算子的 op_graph/CMakeLists.txt 中添加 add_graph_plugin_sources() 函数调用。受影响的算子： - apply_rotary_pos_emb - attention_update - interleave_rope - rotary_position_embedding - rotary_position_embedding_grad - ring_attention_update ## 关联的Issue Fixes #2387 ## 测试原 MR #4905 已通过 CI 流水线验证（ci-pipeline-passed、api-check-pass）。 ## 文档更新无 ## 类型标签 - [x] 🐛 Bug 修复 See merge request: cann/ops-transformer!5231	3 个月前
op_host	fix: 修正 attention_update CheckShape_95 日志中错误变量名 lse → localOut Co-authored-by: wangxun21<wangxun21@huawei.com> # message auto-generated for no-merge-commit merge: !5863 merge fix/attention-update-log-msg into master fix: 修正 attention_update CheckShape_95 日志中错误变量名 lse → localOut Created-by: wangxun21 Commit-by: wangxun21 Merged-by: cann-robot Description: ## 描述修正 attention_update 算子 op_api 中 CheckShape_95 函数的日志错误信息。在 aclnn_attention_update.cpp 第212行，OP_CHECK 校验 goShape.GetDimNum() 时，错误日志描述为 "Dim num of lse should be %ld"，但实际校验的对象是 goShape（即 localOut），而非 lse。将日志中的 "lse" 修正为 "localOut"，使错误提示与实际校验变量一致。 ### 改动原因错误日志中变量名与实际校验对象不一致，会导致用户在遇到参数校验失败时被误导，误以为是 lse 维度问题而实际是 localOut 维度问题。 ### 改动方法将第212行 OP_LOGE 中的 "lse" 替换为 "localOut"。 ## 关联的Issue - #2650 ## 测试该修改仅涉及日志文案，不影响计算逻辑，通过代码审查即可验证。 ## 文档更新无需文档更新。 ## 类型标签 - [x] Bug 修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 重构 - [ ] 测试 - [ ] 构建/CI - [ ] 配置变更 - [ ] 文档更新 - [ ] 依赖升级 - [ ] 安全修复 - [ ] 代码清理 - [ ] 其他 See merge request: cann/ops-transformer!5863	3 个月前
op_kernel	[kernel] attention_update process INF in lse Co-authored-by: echoes_00444757<luwei44@huawei.com> # message auto-generated for no-merge-commit merge: !3732 merge master into master [kernel] attention_update process INF in lse Created-by: luwei_1104 Commit-by: echoes_00444757 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] 🐛 Bug 修复 - [ ] ✨ 新特性 - [ ] ⚡ 性能优化 - [ ] ♻️ 重构 - [ ] 🧪 测试 - [ ] 📦 构建/CI - [ ] 🔧 配置变更 - [ ] 📝 文档更新 - [ ] ⬆️ 依赖升级 - [ ] 🔒 安全修复 - [ ] 🧹 代码清理 - [ ] ❓ 其他，请描述： See merge request: cann/ops-transformer!3732	4 个月前
tests	add atk case Co-authored-by: zerosaki_admin<chenzhaominyi1@huawei.com> # message auto-generated for no-merge-commit merge: !5323 merge master into master add atk file Created-by: zerosaki_admin Commit-by: zerosaki_admin Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 增加前冒烟用例 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#2551](https://gitcode.com/cann/ops-transformer/issues/2551) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 不涉及 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] 🐛 Bug 修复 - [ ] ✨ 新特性 - [ ] ⚡ 性能优化 - [ ] ♻️ 重构 - [x] 🧪 测试 - [ ] 📦 构建/CI - [ ] 🔧 配置变更 - [ ] 📝 文档更新 - [ ] ⬆️ 依赖升级 - [ ] 🔒 安全修复 - [ ] 🧹 代码清理 - [ ] ❓ 其他，请描述： See merge request: cann/ops-transformer!5323	3 个月前
CMakeLists.txt	匹配最新cann版本，上线新版本attention/ffn/moe/posembedding算子 Co-authored-by: huangchuhong<huangchuhong1@h-partners.com> # message auto-generated for no-merge-commit merge: !538 merge master into master 匹配最新cann版本，上线新版本attention/ffn/moe/posembedding算子 Created-by: huang-chuhong Commit-by: huangchuhong Merged-by: cann-robot Description: ## 描述 transformer 同步算子最新开发，涉及attention/ffn/moe/posembedding ## 关联的Issue https://gitcode.com/cann/ops-transformer/issues/315 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-transformer!538	8 个月前
README.md	doc Tools扫描错误修改 Co-authored-by: gitee-yanglulu<yanglulul@h-partners.com> # message auto-generated for no-merge-commit merge: !3348 merge master into master doc Tools扫描错误修改 Created-by: gitee-yanglulu Commit-by: gitee-yanglulu Merged-by: cann-robot Description: doc Tools扫描错误修改 See merge request: cann/ops-transformer!3348	5 个月前

AttentionUpdate

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×

功能说明

算子功能：将各SP域PA算子的输出的中间结果lse，localOut两个局部变量结果更新成全局结果。
计算公式：输入 $lse_i$ 和 $O_i$ 、输出 $O$ 。

lsemax=maxlseilse_{max} = \text{max}lse_i

\sum_i \text{exp}(lse_i - lse_{max})

lsem=lsemax+log(lse)lse_m = lse_{max} + \text{log}(lse)

\sum_i O_i \cdot \text{exp}(lse_i - lse_m)

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
lsei	输入	各SP域的局部lse。	FLOAT32	ND
Oi	输入	各SP域的局部attentionout。	FLOAT32，FLOAT16，BFLOAT16	ND
lsem	输出	更新后的全局lse。	FLOAT32	ND
O	输入	更新后的全局attentionout。	FLOAT32，FLOAT16，BFLOAT16	ND

约束说明

Atlas A2 训练系列产品/Atlas A2 推理系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品：支持FLOAT32、FLOAT16、BFLOAT16的Oi和O。
Ascend 950PR/Ascend 950DT：支持FLOAT32、FLOAT16、BFLOAT16的Oi和O，且Oi和O数据类型相同。
序列并行的并行度sp取值范围[1, 16]。
headDim取值范围[8, 512]且是8的倍数。
不支持非连续的Tensor。
支持空Tensor。

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_attention_update	通过`aclnnAttentionUpdate`接口方式调用AttentionUpdate算子。