Star0

Star0

cann-robotfix(global_lp_pool):初始化workspace size, 修复arch35异常(#5504 )

425ca418创建于 11 天前历史提交

文件	最后提交记录	最后更新时间
examples	feat(pooling): add GlobalLpPool operator Co-authored-by: weixin_51153241<fanjie43@huawei.com> # message auto-generated for no-merge-commit merge: !6398 merge feature/add-global-lp-pool into master feat(pooling): add GlobalLpPool operator Created-by: weixin_51153241 Commit-by: weixin_51153241 Merged-by: cann-robot Description: ## 算子说明 Global LP Pool — 计算输入张量空间维度的 Lp 范数: y = (sum(\|x\|^p))^(1/p), keepdims=True ## 支持规格 - 输入: 4D (NCHW) 或 5D (NCD0D1D2) - DTypes: FP32, FP16, BF16 - 属性: p (float, 默认 2.0) - 架构: ascend950/arch35 - 接口: aclnn_exclude (直调) ## 文件清单 (20 files) - op_kernel/ — VF 级 kernel 实现 (双缓冲 + 二分归约树 + \|x\|^p + (1/p) root) - op_host/ — Tiling (ubFactor, 多核切分, cacheBuf 约束, empty tensor 分支) - op_graph/ — Proto 定义 - tests/ — Tiling 单测 (5 用例: 4D/5D/大通道/空张量) - README.md — 算子说明 See merge request: cann/ops-nn!6398	2 个月前
op_graph	4个算子原型增去重宏 Co-authored-by: zhoulong50_hw<zhoulong50@huawei.com> # message auto-generated for no-merge-commit merge: !9213 merge proto_fix into master 4个算子原型增去重宏 Created-by: zhoulong50_hw Commit-by: zhoulong50_hw Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 1. AscendRequant 2. GlobalLpPool 3. AscendDequantS16 4. AscendRequantS16 4 个算子增加原型去重宏。通过增加 #define OPS_PROTO_DEF_OPNAME 宏的形式。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5177 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 二级冒烟 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 无 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9213	20 天前
op_host	fix(global_lp_pool):初始化workspace size, 修复arch35异常(#5504) Co-authored-by: 2637309376@qq.com<lixin849@huawei.com> # message auto-generated for no-merge-commit merge: !9774 merge fix-global-lp-pool-workspace into master fix(global_lp_pool):初始化workspace size, 修复arch35异常(#5504) Created-by: weixin_68109557 Commit-by: 2637309376@qq.com Merged-by: cann-robot Description: ## 描述 Bug修复：global_lp_pool arch35 workspace size 初始化。 ## 关联的Issue 关联 Issue：#5504 ## 测试 CI 流水线已全部通过。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9774	11 天前
op_kernel	feat(pooling): add GlobalLpPool operator Co-authored-by: weixin_51153241<fanjie43@huawei.com> # message auto-generated for no-merge-commit merge: !6398 merge feature/add-global-lp-pool into master feat(pooling): add GlobalLpPool operator Created-by: weixin_51153241 Commit-by: weixin_51153241 Merged-by: cann-robot Description: ## 算子说明 Global LP Pool — 计算输入张量空间维度的 Lp 范数: y = (sum(\|x\|^p))^(1/p), keepdims=True ## 支持规格 - 输入: 4D (NCHW) 或 5D (NCD0D1D2) - DTypes: FP32, FP16, BF16 - 属性: p (float, 默认 2.0) - 架构: ascend950/arch35 - 接口: aclnn_exclude (直调) ## 文件清单 (20 files) - op_kernel/ — VF 级 kernel 实现 (双缓冲 + 二分归约树 + \|x\|^p + (1/p) root) - op_host/ — Tiling (ubFactor, 多核切分, cacheBuf 约束, empty tensor 分支) - op_graph/ — Proto 定义 - tests/ — Tiling 单测 (5 用例: 4D/5D/大通道/空张量) - README.md — 算子说明 See merge request: cann/ops-nn!6398	2 个月前
tests	feat(pooling): add GlobalLpPool operator Co-authored-by: weixin_51153241<fanjie43@huawei.com> # message auto-generated for no-merge-commit merge: !6398 merge feature/add-global-lp-pool into master feat(pooling): add GlobalLpPool operator Created-by: weixin_51153241 Commit-by: weixin_51153241 Merged-by: cann-robot Description: ## 算子说明 Global LP Pool — 计算输入张量空间维度的 Lp 范数: y = (sum(\|x\|^p))^(1/p), keepdims=True ## 支持规格 - 输入: 4D (NCHW) 或 5D (NCD0D1D2) - DTypes: FP32, FP16, BF16 - 属性: p (float, 默认 2.0) - 架构: ascend950/arch35 - 接口: aclnn_exclude (直调) ## 文件清单 (20 files) - op_kernel/ — VF 级 kernel 实现 (双缓冲 + 二分归约树 + \|x\|^p + (1/p) root) - op_host/ — Tiling (ubFactor, 多核切分, cacheBuf 约束, empty tensor 分支) - op_graph/ — Proto 定义 - tests/ — Tiling 单测 (5 用例: 4D/5D/大通道/空张量) - README.md — 算子说明 See merge request: cann/ops-nn!6398	2 个月前
CMakeLists.txt	feat(pooling): add GlobalLpPool operator Co-authored-by: weixin_51153241<fanjie43@huawei.com> # message auto-generated for no-merge-commit merge: !6398 merge feature/add-global-lp-pool into master feat(pooling): add GlobalLpPool operator Created-by: weixin_51153241 Commit-by: weixin_51153241 Merged-by: cann-robot Description: ## 算子说明 Global LP Pool — 计算输入张量空间维度的 Lp 范数: y = (sum(\|x\|^p))^(1/p), keepdims=True ## 支持规格 - 输入: 4D (NCHW) 或 5D (NCD0D1D2) - DTypes: FP32, FP16, BF16 - 属性: p (float, 默认 2.0) - 架构: ascend950/arch35 - 接口: aclnn_exclude (直调) ## 文件清单 (20 files) - op_kernel/ — VF 级 kernel 实现 (双缓冲 + 二分归约树 + \|x\|^p + (1/p) root) - op_host/ — Tiling (ubFactor, 多核切分, cacheBuf 约束, empty tensor 分支) - op_graph/ — Proto 定义 - tests/ — Tiling 单测 (5 用例: 4D/5D/大通道/空张量) - README.md — 算子说明 See merge request: cann/ops-nn!6398	2 个月前
README.md	feat(pooling): add GlobalLpPool operator Co-authored-by: weixin_51153241<fanjie43@huawei.com> # message auto-generated for no-merge-commit merge: !6398 merge feature/add-global-lp-pool into master feat(pooling): add GlobalLpPool operator Created-by: weixin_51153241 Commit-by: weixin_51153241 Merged-by: cann-robot Description: ## 算子说明 Global LP Pool — 计算输入张量空间维度的 Lp 范数: y = (sum(\|x\|^p))^(1/p), keepdims=True ## 支持规格 - 输入: 4D (NCHW) 或 5D (NCD0D1D2) - DTypes: FP32, FP16, BF16 - 属性: p (float, 默认 2.0) - 架构: ascend950/arch35 - 接口: aclnn_exclude (直调) ## 文件清单 (20 files) - op_kernel/ — VF 级 kernel 实现 (双缓冲 + 二分归约树 + \|x\|^p + (1/p) root) - op_host/ — Tiling (ubFactor, 多核切分, cacheBuf 约束, empty tensor 分支) - op_graph/ — Proto 定义 - tests/ — Tiling 单测 (5 用例: 4D/5D/大通道/空张量) - README.md — 算子说明 See merge request: cann/ops-nn!6398	2 个月前

GlobalLpPool

产品支持情况

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

功能说明

算子功能：对输入张量的空间维度进行全局Lp池化计算，公式为y = (sum(|x|^p))^(1/p)，保持空间维度为1（keepdims=True）。支持4D(NCHW)和5D(NCD0D1D2)输入。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待进行global_lp_pool计算的输入张量，4D(NCHW)或5D(NCD0D1D2)。	FLOAT、FLOAT16、BFLOAT16	ND
p	属性	Lp范数的指数p。默认值为2.0（即L2范数）。	FLOAT	-
y	输出	全局Lp池化结果。空间维度归约为1，与输入x同dtype。	FLOAT、FLOAT16、BFLOAT16	ND

约束说明

输入张量维度仅支持4D(NCHW)或5D(NCD0D1D2)。
p的取值范围建议为正浮点数（p>0）。p=0时行为未定义，不建议使用。

调用说明

调用方式	样例代码	说明
图模式接口	test_geir_global_lp_pool	通过算子IR构图方式调用GlobalLpPool算子。