| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
[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 | 5 天前 | |
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 | 29 天前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 5 天前 | ||
| 29 天前 |