| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
test: 为 activation 和 norm 算子补充ut测试 Co-authored-by: cai-chengchao<caichengchao1@h-partners.com> # message auto-generated for no-merge-commit merge: !4312 merge ut_0427 into master test: 为 activation 和 norm 算子补充ut测试 Created-by: cai-chengchao Commit-by: cai-chengchao Merged-by: cann-robot Description: ## 描述 为 activation 和 norm 目录下的算子补充ut测试，提升测试覆盖率。包括 op_kernel 和 op_host 两级测试，覆盖多种数据类型和不同维度组合。 ### 改动详情 **新增测试文件统计**： - op_kernel 级别 APT 测试：15 个算子，共 45 个文件（CMakeLists.txt、tiling_def.h、test_*.apt.cpp） - op_host 级别 infershape 测试：3 个算子，共 3 个文件 **涉及的算子**： activation 目录： - confusion_softmax_grad - log_softmax_grad - log_softmax_v2 - softmax_grad - softmax_v2 norm 目录： - add_rms_norm_dynamic_mx_quant - batch_norm - group_norm_v2 - layer_norm - layer_norm_v3 - lp_norm_v2 - rms_norm_dynamic_mx_quant - rms_norm_quant_v2 - sync_batch_norm_backward_elemt - sync_batch_norm_backward_reduce - sync_batch_norm_gather_stats **其他改动**： - 修复 infershape 文件缺失头文件：为 add_rms_norm_dynamic_mx_quant 和 rms_norm_quant_v2 的 infershape 文件添加 #include <algorithm> 头文件引用 **测试覆盖**： - op_kernel APT 测试：覆盖 float、half 等数据类型，不同 tiling key 场景（如 small_r、recompute、ara 等） - op_host infershape 测试：覆盖 DT_FLOAT16、DT_FLOAT、DT_BF16 数据类型，多种维度组合（1D-8D），包括动态 shape（-2） ## 关联的Issue 关联 Issue #2452 ## 测试 所有新增测试文件均可通过编译和执行： - CI Pipeline 已通过验证 - 测试用例覆盖多种数据类型和 shape 组合 ## 文档更新 无文档更新 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!4312 | 3 个月前 | |
[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 天前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 3 个月前 | ||
| 20 天前 | ||
| 20 天前 |