| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
fix: softmax_grad 二分累加替换为高阶API ReduceSum 解决精度问题 Co-authored-by: 朱子贤<zhuzixian1@h-partners.com> # message auto-generated for no-merge-commit merge: !5348 merge softmaxgrad into master fix: softmax_grad 二分累加替换为高阶API ReduceSum 解决精度问题 Created-by: zhuzixian-lr Commit-by: 朱子贤 Merged-by: cann-robot Description: ## 描述 将 softmax_grad 算子中手写的二分累加(LastReduceSum/LastReduceSumSmallR)替换为高阶API AscendC::ReduceSum<float, AscendC::Pattern::Reduce::AR, true>,解决当归约长度超过16384时的精度问题。 ### 改动原因 手写的二分累加实现 LastReduceSum 采用逐级折叠方式计算归约求和,其最大支持长度为 16384(2^14)。当 softmax_grad 的归约维度长度超过此限制时,累加结果出现精度偏差,导致算子输出不准确。 ### 改动方法 1. **kernel 层**:删除 LastReduceSum 和 LastReduceSumSmallR 两个手写二分累加函数,改用 AscendC::ReduceSum<float, AscendC::Pattern::Reduce::AR, true> 高阶API 实现归约求和 2. **kernel 层**:移除 reduceSumTempTensor_ 临时缓冲区及 AR_RECOMPUTE_BINARY_TMP_BTYES 常量(高阶API不再需要额外中间缓冲) 3. **tiling 层**:从 UB 可用空间计算中移除 AR_RECOMPUTE_BINARY_TMP_BTYES,释放更多 UB 空间供计算使用 ## 关联的Issue - #2938 ## 测试 - 验证归约长度 ≤ 16384 时精度不变 - 验证归约长度 > 16384 时精度满足要求 - 验证不同 dtype(FP16/FP32/BF16)下的计算结果正确性 ## 文档更新 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!5348 | 2 个月前 | |
remove simplified_key, default value is 0 Co-authored-by: cong-jiyu<congjiyu@h-partners.com> # message auto-generated for no-merge-commit merge: !4892 merge master into master remove simplified_key, default value is 0 Created-by: cong-jiyu Commit-by: cong-jiyu Merged-by: cann-robot Description: ## 描述 当simplified_key中属性只有default=0,且ascendc_config中配置了算子,那么simplified_key.ini文件可以删除,因为默认值为0. 修改了ascendc_config.json文件,将算子host/config中配置了soc,但是未在ascendc_config.json中的算子补充上去。 ## 关联的Issue [#1985](https://gitcode.com/cann/ops-nn/issues/1985) ## 测试 编译950、910bpass ## 文档更新 更新了ascendc_config.json配置了算子的simplified_key的ini文件 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!4892 | 3 个月前 | |
Arch编码更新 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !1116 merge ut_test into master Arch编码更新 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 Arch编码更新 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!1116 | 6 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 2 个月前 | ||
| 3 个月前 | ||
| 6 个月前 |