| MaxPoolWithArgmax 新增算子 及 部分pool算子头文件补充 Co-authored-by: 王世龙<wangshilong23@huawei.com> # message auto-generated for no-merge-commit merge: !1495 merge vec_mpwa into master MaxPoolWithArgmax 新增算子 及 部分pool算子头文件补充 Created-by: wang-shilong32 Commit-by: 王世龙 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> - 新增在Ascend 950PR/Ascend 950DT上支持MaxPoolWithArgmax算子AICore实现。 - 包含算子原型、文档、tiling kernel实现代码、UT等。 - 支持范围：单算子、geir。 - 主要方案： 1. 大kernel模板：即C维度大小大于VL长度一半。 2. 小kernel模板: 即C维度大小小于等于VL长度一半。 3. SIMT模板：NHWC和NCHW格式的兜底模板，dilation !=1等场景。 - 同时补充其他算子AvgPool3d、MaxPool3dGradWithArgmax、MaxPoolV3芯片校验及相关头文件引用。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/793 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 本地验证精度功能已通过，冒烟已通过。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!1495 | 6 个月前 |
