ops-nn/norm/batch_norm/op_host · CANN/ops-nn

cann-robotfix(batch_norm): 修复 infer 模板冗余写及 small_ab1 输出错误

文件	最后提交记录	最后更新时间
arch35	fix(batch_norm): 修复 infer 模板冗余写及 small_ab1 输出错误 Co-authored-by: qianzehong<qianzehong@huawei.com> # message auto-generated for no-merge-commit merge: !5962 merge fix/batch_norm_remove_redundant_meanvar_write into master fix(batch_norm): 修复 infer 模板冗余写及 small_ab1 输出错误 Created-by: qianzehong Commit-by: qianzehong Merged-by: cann-robot Description: ## 描述 batch_norm 算子的 3 个 infer 新模板存在两类问题： ### 问题 1：冗余写 continuous_a、small_a、small_ab1 三个模板中，`CopyOutMeanVar` 被所有核无条件调用，导致多个核向同一 GM 地址写同一份数据（batch_mean、batch_variance、reserve_space_1、reserve_space_2）。修复：在 `CopyOutMeanVar` 函数中增加 `GetBlockIdx() == 0` 守卫，只让 core 0 写。 ### 问题 2：small_ab1 输出错误 small_ab1 模板中，`VFPrepareSmallAB1ParamCache` 使用 `DataCopyGather` 读取 mean/var 时，由于 `paramCacheElemLen`（128）大于 `meanQueue_` buffer 大小（totalALen=4 个元素），导致 meanLocal 被破坏，batch_mean 和 reserve_space_1 输出错误数据（epsilon 值而非实际 mean）。修复： - 单独申请 `meanOutQueue_` 和 `varOutQueue_`（`TQueBind` 类型，BUFFER_NUM=1） - 从 GM 重新加载 mean/var 后再写出到输出 GM 地址 - 添加正确的 `EnQue`/`DeQue` 操作 - tiling 中 `ubCanUseBytes` 计算减去这两个 buffer 的空间老模板 `batch_norm_infer_last_channel.h` 已有 `needCopyOut` 条件判断，不受影响。 ## 修改文件 \| 文件 \| 修改内容 \| \|------\|----------\| \| `batch_norm_infer_last_channel_continuous_a.h` \| 添加 `GetBlockIdx()` 守卫 \| \| `batch_norm_infer_last_channel_small_a.h` \| 添加 `GetBlockIdx()` 守卫 \| \| `batch_norm_infer_small_ab1.h` \| 添加守卫 + 单独 buffer 修复输出错误 \| \| `batch_norm_tiling_infer_arch35.cpp` \| tiling 空间计算减去新增 buffer \| ## 关联的Issue - #3290 ## 测试 - TTK kernel 测试 fp32 NHWC [256,165,165,4]（small_a 902000）：5 个有效输出 100% PASS - TTK kernel 测试 fp16 NHWC [256,165,165,192]（continuous_a 901000）：5 个有效输出 100% PASS - TTK kernel 测试 fp32 NCHW [256,4,2,2]（small_ab1 911000）：5 个有效输出 100% PASS ## 文档更新无需更新文档 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他 See merge request: cann/ops-nn!5962	1 个月前
config	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	2 个月前
batch_norm_def.cpp	删除开源仓中的mc_xx的信息库 Co-authored-by: xufeng12121<1074805447@qq.com> # message auto-generated for no-merge-commit merge: !4814 merge mov into master 删除开源仓中的mc_xx的信息库 Created-by: xufeng12121 Commit-by: xufeng12121 Merged-by: cann-robot Description: ## 描述删除开源仓中的mc_xx的信息库. 对应信息库会迁移到modext仓 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/2688 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 1.本地单算子验证 2.二级冒烟：http://7.215.10.221/scheduler/alljobs?length=100&isalljobs=true&search=20260514_091630371 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!4814	2 个月前
batch_norm_infershape.cpp	BatchNorm/ConfusionSoftmaxGrad支持下一代芯片 Co-authored-by: cai-chengchao<caichengchao1@h-partners.com> # message auto-generated for no-merge-commit merge: !2550 merge support_next_soc into master BatchNorm/ConfusionSoftmaxGrad支持下一代芯片 Created-by: cai-chengchao Commit-by: cai-chengchao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> BatchNorm/ConfusionSoftmaxGrad支持下一代芯片 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/1579 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ut、st、二级冒烟测试通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 新增了norm/batch_norm/README.md文件新增了norm/confusion_softmax_grad/README.md文件 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：BatchNorm/ConfusionSoftmaxGrad支持下一代芯片 See merge request: cann/ops-nn!2550	4 个月前