| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
format cpp Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !6784 merge issue_fix into master format cpp Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 批量刷新cpp代码格式 ## 关联的Issue [#3791](https://gitcode.com/cann/ops-nn/issues/3791) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:代码格式化 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!6784 | 1 个月前 | |
[Fix] Fix softmax family AR full_load binaryTmpLocalTensor buffer overflow Co-authored-by: zhuzixian-lr<zhuzixian1@h-partners.com> # message auto-generated for no-merge-commit merge: !7630 merge fix/softmax-grad-ar-binary-tmp-buffer into master [Fix] Fix softmax family AR full_load binaryTmpLocalTensor buffer overflow Created-by: zhuzixian-lr Commit-by: zhuzixian-lr Merged-by: cann-robot Description: ## 问题 softmax_grad、log_softmax_grad、confusion_softmax_grad 的 AR full_load tiling 中 BINARY_TMP_LOCAL_SHAPE = 512 写死,对应 kernel 中 AR_FULL_LOAD_BINARY_TMP_BYTES = 512。 当 r > 2*VL_FP32 时,buffer 实际需求为 ubFactor * outerLoopDstStride * sizeof(float),与 r 和 ubFactor 相关,512 字节远不够(最严重场景溢出 6.9 倍)。 ## 修改 1. **softmax_grad_tiling.h**: SoftmaxGradARTilingData 新增 binaryTmpSize 字段(影响 softmax_grad、log_softmax_grad、confusion_softmax_grad 三个算子) 2. **softmax_grad_tiling_ar_full_load.cpp**: 删除写死512,新增 FindNearestPower2,动态计算 binaryTmpPerRow,将 binaryTmpPerRow 放入 ubFactor 分母 3. **softmax_grad/log_softmax_grad/confusion_softmax_grad kernel**: 删除写死常量,使用 tl_->binaryTmpSize 4. **log_softmax_v2 kernel**: 删除写死常量,使用 tl_->binaryTmpSize(tiling 已由 #7550 修复) 5. 同步修改所有 UT 文件(op_host tiling data size、op_kernel tiling_def、test case) ## 验证 - 编译4个算子出包安装成功 - TTK测试13用例全部 MEM=PASS,TilingKey=1000(AR full_load) - torch_npu精度验证:softmax_grad/log_softmax_v2/log_softmax_grad 全部 PASS - op_host UT 20 tests PASSED - op_kernel UT 18 tests PASSED 关联PR: #7550 See merge request: cann/ops-nn!7630 | 20 天前 | |
[Fix] Fix softmax family AR full_load binaryTmpLocalTensor buffer overflow Co-authored-by: zhuzixian-lr<zhuzixian1@h-partners.com> # message auto-generated for no-merge-commit merge: !7630 merge fix/softmax-grad-ar-binary-tmp-buffer into master [Fix] Fix softmax family AR full_load binaryTmpLocalTensor buffer overflow Created-by: zhuzixian-lr Commit-by: zhuzixian-lr Merged-by: cann-robot Description: ## 问题 softmax_grad、log_softmax_grad、confusion_softmax_grad 的 AR full_load tiling 中 BINARY_TMP_LOCAL_SHAPE = 512 写死,对应 kernel 中 AR_FULL_LOAD_BINARY_TMP_BYTES = 512。 当 r > 2*VL_FP32 时,buffer 实际需求为 ubFactor * outerLoopDstStride * sizeof(float),与 r 和 ubFactor 相关,512 字节远不够(最严重场景溢出 6.9 倍)。 ## 修改 1. **softmax_grad_tiling.h**: SoftmaxGradARTilingData 新增 binaryTmpSize 字段(影响 softmax_grad、log_softmax_grad、confusion_softmax_grad 三个算子) 2. **softmax_grad_tiling_ar_full_load.cpp**: 删除写死512,新增 FindNearestPower2,动态计算 binaryTmpPerRow,将 binaryTmpPerRow 放入 ubFactor 分母 3. **softmax_grad/log_softmax_grad/confusion_softmax_grad kernel**: 删除写死常量,使用 tl_->binaryTmpSize 4. **log_softmax_v2 kernel**: 删除写死常量,使用 tl_->binaryTmpSize(tiling 已由 #7550 修复) 5. 同步修改所有 UT 文件(op_host tiling data size、op_kernel tiling_def、test case) ## 验证 - 编译4个算子出包安装成功 - TTK测试13用例全部 MEM=PASS,TilingKey=1000(AR full_load) - torch_npu精度验证:softmax_grad/log_softmax_v2/log_softmax_grad 全部 PASS - op_host UT 20 tests PASSED - op_kernel UT 18 tests PASSED 关联PR: #7550 See merge request: cann/ops-nn!7630 | 20 天前 | |
add ut cmake for softmax_grad Co-authored-by: chensi888<chensi79@huawei.com> # message auto-generated for no-merge-commit merge: !894 merge master into master add ut cmake for softmax_grad Created-by: chensi79 Commit-by: chensi888 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-nn!894 | 6 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 1 个月前 | ||
| 20 天前 | ||
| 20 天前 | ||
| 6 个月前 |