Star0

Star0

cann-robotoptim、pooling打标签

e3d14dd4创建于 8月6日历史提交

文件	最后提交记录	最后更新时间
docs	optim、pooling打标签 Co-authored-by: y60124828<yanglou1@h-partners.com> # message auto-generated for no-merge-commit merge: !8311 merge master into master optim、pooling打标签 Created-by: yanglu-1 Commit-by: y60124828 Merged-by: cann-robot Description: ## 描述 optim、pooling打标签 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!8311	1 个月前
examples	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	2 个月前
op_api	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	2 个月前
op_host	refactor: 统一 arch32 → arch22 命名（index/pooling/conv） Co-authored-by: chen-xing-yu<chenxingyu18@huawei.com> # message auto-generated for no-merge-commit merge: !8210 merge refactor/arch32_to_arch22 into master refactor: 统一 arch32 → arch22 命名（index/pooling/conv） Created-by: chenxingyu18 Commit-by: chen-xing-yu Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 将 index/、pooling/、conv/ 下所有 arch32 目录与引用统一改为 arch22，与全仓主流命名约定对齐（arch22 用于 ascend910b/910_93，arch35 用于 ascend950）。 - 目录重命名 arch32 → arch22（op_host/op_tiling、op_kernel、tests/ut 等） - 文件内容：include 路径、CMakeLists 的 SUPPORT_TILING_DIR、注释中的 arch32 → arch22 - arch35 目录与引用未受影响 - 顺带修复两个被禁用（main() 提前 return 0）的 eager example：index_add、scatter_nd_update ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> #4493 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 基于 ascend910b（CANN 9.1.0 / 910B3）完成统一打包构建 + eager 验证，涉及算子 index_fill、scatter_nd_update、masked_scatter、adaptive_max_pool3d(_grad)、max_pool3d_grad_with_argmax、avg_pool3_d、inplace_index_add_with_sorted、conv3d_backprop_filter_v2、conv3d_backprop_input_v2、conv3d_transpose_v2 均验证通过；并用改名测试（移除 vendor kernel 后 example 失败）确认跑的是改名后的实现而非 stock CANN。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：命名统一（arch32 → arch22） ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!8210	1 个月前
op_kernel	refactor: 统一 arch32 → arch22 命名（index/pooling/conv） Co-authored-by: chen-xing-yu<chenxingyu18@huawei.com> # message auto-generated for no-merge-commit merge: !8210 merge refactor/arch32_to_arch22 into master refactor: 统一 arch32 → arch22 命名（index/pooling/conv） Created-by: chenxingyu18 Commit-by: chen-xing-yu Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 将 index/、pooling/、conv/ 下所有 arch32 目录与引用统一改为 arch22，与全仓主流命名约定对齐（arch22 用于 ascend910b/910_93，arch35 用于 ascend950）。 - 目录重命名 arch32 → arch22（op_host/op_tiling、op_kernel、tests/ut 等） - 文件内容：include 路径、CMakeLists 的 SUPPORT_TILING_DIR、注释中的 arch32 → arch22 - arch35 目录与引用未受影响 - 顺带修复两个被禁用（main() 提前 return 0）的 eager example：index_add、scatter_nd_update ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> #4493 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 基于 ascend910b（CANN 9.1.0 / 910B3）完成统一打包构建 + eager 验证，涉及算子 index_fill、scatter_nd_update、masked_scatter、adaptive_max_pool3d(_grad)、max_pool3d_grad_with_argmax、avg_pool3_d、inplace_index_add_with_sorted、conv3d_backprop_filter_v2、conv3d_backprop_input_v2、conv3d_transpose_v2 均验证通过；并用改名测试（移除 vendor kernel 后 example 失败）确认跑的是改名后的实现而非 stock CANN。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：命名统一（arch32 → arch22） ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!8210	1 个月前
tests	pooling UT delete dlog_setlevel Co-authored-by: liuhongpeng<liuhongpeng2@h-partners.com> # message auto-generated for no-merge-commit merge: !7995 merge dlog into master pooling UT delete dlog_setlevel Created-by: liu_hp711 Commit-by: liuhongpeng Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 删除掉pooling目录下的UT使用的dlog_setlevel函数。解决上库UT报错的问题。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/4364 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 运行UT，冒烟。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!7995	1 个月前
CMakeLists.txt	AddMaxPool3DGrad Co-authored-by: huohuo_wy<wangyan389@huawei.com> # message auto-generated for no-merge-commit merge: !799 merge pooling into master AddMaxPool3DGrad Created-by: huohuo_wangyan Commit-by: huohuo_wy Merged-by: cann-robot Description: ## 描述 1) 新增算子实现：通过两段式接口（获取工作空间与执行）实现了上述反向传播算子，包含严格的参数校验、张量格式转换及智能选择底层计算内核等逻辑。2) 代码重构与优化：对 AdaptiveMaxPool3DGrad 和 MaxPool3DGradWithArgmax 等算子的内核及分片（Tiling）实现进行了深度重构，通过引入公共函数库和模板基类（如 pool_3d_common）大幅消除了代码重复，统一了计算逻辑，并修复了输出维度计算、缓冲区初始化等潜在缺陷。3) 文档与构建完善：全面更新了相关算子的API文档，细化了产品支持列表，补充了结构化参数说明和完整调用示例；简化了多个项目的CMake构建逻辑，提升了构建的清晰度和可维护性。4) 测试增强：修正了测试代码中的路径与变量错误，新增了针对重叠模式等边界场景的单元测试，提升了测试的健壮性和覆盖率。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!799	7 个月前
README.md	文档修正: adaptive_max_pool3d_grad README 950支持状态修正 Co-authored-by: huyihang<huyihang6@h-partners.com> # message auto-generated for no-merge-commit merge: !5814 merge a5doc into master 文档修正: adaptive_max_pool3d_grad README 950支持状态修正 Created-by: huyihang1117 Commit-by: huyihang Merged-by: cann-robot Description: ## 描述 [ops-nn] 文档修正: adaptive_max_pool3d_grad README 950支持状态修正 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/3162 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 无 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> pooling/adaptive_max_pool3d_grad/README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!5814	3 个月前

AdaptiveMaxPool3DGrad

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×

功能说明

算子功能：正向自适应最大池化的反向传播，将梯度回填到每个自适应窗口最大值的坐标处，相同坐标处累加。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待进行AdaptiveMaxPool3DGrad计算的入参。	FLOAT、FLOAT16、BFLOAT16	NCDHW
grad	输入	待进行AdaptiveMaxPool3DGrad计算的入参，表示当前节点的梯度。	FLOAT、FLOAT16、BFLOAT16	NCDHW
argmax	输入	待进行AdaptiveMaxPool3DGrad计算的入参。表示正向输入中最大元素的索引位置。数据格式、shape与入参`grad`的保持一致。	INT32	NCDHW
y	输出	待进行AdaptiveMaxPool3DGrad计算的出参。数据格式、shape与入参`x`的保持一致。	FLOAT、FLOAT16、BFLOAT16	NCDHW

约束说明

无

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_adaptive_max_pool3d_backward.cpp	通过aclnnAdaptiveMaxPool3dBackward接口方式调用AdaptiveMaxPool3DGrad算子。