| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
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 | 3 个月前 | |
feat: 新增 AdaptiveMaxPool2d AICPU 算子 Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !8490 merge dev into master feat: 新增 AdaptiveMaxPool2d AICPU 算子 Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 本 PR 新增 AdaptiveMaxPool2d AICPU 算子，对输入信号进行 2D 自适应最大池化，输出尺寸为 H x W，兼容 PyTorch 的 AdaptiveMaxPool2d 算子。 主要变更： 1. **算子原型注册**：adaptive_max_pool2d_proto.h 通过 REG_OP(AdaptiveMaxPool2d) 注册算子，定义输入 x、输出 y/argmax、属性 output_size 2. **算子定义**：adaptive_max_pool2d_aicpu_def.cpp 通过 OpDef 注册算子定义，配置 AICPU 调度 + format agnostic 3. **AICPU Kernel 实现**：adaptive_max_pool2d_aicpu.cpp 实现核心计算逻辑——按 output_size 自动计算每个输出像素对应的输入区域 [StartIndex, EndIndex)，在区域内找最大值及其索引，使用 ParallelFor 对 batch/channel 维并行加速 4. **数据类型与格式**：支持 float16/float32/float64 输入 + int32/int64 索引输出，支持 NCHW/NHWC 格式 + 3D/4D 输入 5. **UT**：test_adaptive_max_pool2d.cpp 覆盖各数据类型、输入格式、3D/4D 场景 6. **Example**：test_geir_adaptive_max_pool2d.cpp 端到端调用样例 7. **文档**：op_list.md 补充算子说明 ## 关联的Issue #4698 ## 测试 1. **单元测试 - 数据类型覆盖** - 执行 test_adaptive_max_pool2d，验证 float16/float32/float64 三种数据类型计算结果正确 - 验证 argmax 索引输出为 int32/int64 两种类型 2. **单元测试 - 输入格式与维度** - 验证 3D (CHW/HWC) 和 4D (NCHW/NHWC) 输入场景 - 验证 output_size 为单元素列表（正方形输出）和双元素列表（矩形输出） 3. **Example 端到端验证** - 执行 test_geir_adaptive_max_pool2d，验证算子端到端调用流程正常 4. **编译构建测试** - 验证 op_graph 和 op_kernel_aicpu 的 CMakeLists.txt 配置正确，算子能正确构建 ## 文档更新 - docs/zh/op_list.md：新增 AdaptiveMaxPool2d 算子说明条目 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [ ] AI辅助编写 See merge request: cann/ops-nn!8490 | 1 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 3 个月前 | ||
| 1 个月前 |