Thanks for your pull-request.
The full list of commands accepted by me can be found at here。
You can get sig-info at here

PR Approval Progress

✅ Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

💡 Tip:

• Committer can comment /approve or /lgtm
• Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

weixin_51153241, thanks for your pull request. All authors of the commits have signed the CLA. 👍

变更摘要

本 PR 为 CANN 算子库新增 GlobalLpPool 算子,实现输入张量空间维度的 Lp 范数池化(y = (sum(|x|^p))^(1/p), keepdims=True)。涵盖 4D(NCHW)和 5D(NCD0D1D2)输入、FP32/FP16/BF16 三种数据类型,基于 ascend950/arch35 架构。核心组件包括:tiling 参数计算(global_lp_pool_tiling_arch35.cpp)、VF 级 kernel 实现(双缓冲 + 二分归约树,global_lp_pool_kernel.h)、算子定义与 shape 推导、JSON 配置、proto 图注册,以及 4 个测试/示例程序和 1 个 tiling 单元测试。

主要改动

• 新增算子核心 kernel 实现:global_lp_pool_kernel.h 包含模板类 GlobalLpPoolKernel,实现双缓冲 CopyIn、VF 指令集 Cast/Power/ReduceSum、二分归约树 Accumulate(GetCacheID/DoCachingVf)、以及 PostElewise 逆幂运算(Sqrt/Ln/Muls/Exp),支持 isEmptyTensor 和 isTailR 模板分派。

• 新增 tiling 参数计算:global_lp_pool_tiling_arch35.cpp 实现多步 UB 切分算法(CalcAUbFactor → CalcRFromBudget → TryExpandA → CheckCacheBufConstraint),支持多核切分、空张量路径,并将结果写入 GlobalLpPoolTilingData 结构体。

• 新增算子注册与接口定义:global_lp_pool_def.cpp 通过 OpDef 注册输入/输出/属性(p 默认 2.0),global_lp_pool_proto.h 通过 REG_OP 注册图算子,global_lp_pool_infershape.cpp 实现空间维度归约为 1 的 shape 推导。

• 测试与示例程序:4 个示例文件覆盖 aclnn 两段式调用、200 个 shape×dtype×p 组合、综合精度对比、整数输入精度验证;1 个 tiling 单元测试覆盖基本/大通道/5D/空张量场景。

• 构建与配置:根 CMakeLists.txt 设置 ascend950/arch35 编译目标;global_lp_pool.json / global_lp_pool_binary.json 定义三种 dtype 的 kernel 二进制映射;tiling_key.h 定义 isEmptyTensor/isTailR 模板参数选择。

审查覆盖的文件(全部 23 个):CMakeLists.txt(无问题),README.md(无问题),test_200.cpp(P2 泄漏),test_aclnn_global_lp_pool.cpp(P2 泄漏),test_comprehensive.cpp(P2 泄漏 + P3 死代码),test_integer.cpp(P2 泄漏),op_graph/CMakeLists.txt(无问题),global_lp_pool_proto.h(无问题),global_lp_pool_tiling_arch35.cpp(P3 调试日志),global_lp_pool_tiling_arch35.h(无问题),global_lp_pool.json(无问题),global_lp_pool_binary.json(无问题),global_lp_pool_def.cpp(无问题),global_lp_pool_infershape.cpp(无问题),global_lp_pool_empty.h(无问题),global_lp_pool_kernel.h(无问题),global_lp_pool_tiling_data.h(无问题),global_lp_pool_tiling_key.h(无问题),global_lp_pool_apt.cpp(无问题),tests/CMakeLists.txt(无问题),tests/ut/CMakeLists.txt(无问题),tests/ut/op_host/CMakeLists.txt(无问题),test_global_lp_pool_tiling.cpp(无问题)。

代码审查

审查总结

共审查 23 个变更文件，发现 7 个问题：

逐文件审查结果：

整体风险评估：低风险。 核心算子逻辑（kernel、tiling、infer shape、proto、op def）未发现正确性或安全性缺陷。所有 P2 问题集中在测试/示例代码的资源管理上，不影响生产运行。P3 问题为构建规范、死代码和文档准确性，均为可选改进项。

💬 仅评论

/approve