Star0

Star0

cann-robotfeat: 支持八个迁移算子的常量折叠(#2551 )

35f36a06创建于 26 天前历史提交

文件	最后提交记录	最后更新时间
docs	docs目录基本概念md文件名改为英文 Co-authored-by: chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !4182 merge master into master docs目录基本概念md文件名改为英文 Created-by: gitcode-chenjiao Commit-by: chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> docs目录基本概念md文件名改为英文：避免link中的中文字符引发的跳转异常，例如两段式接口.md变成%E4%B8%A4%E6%AE%B5%E5%BC%8F%E6%8E%A5%E5%8F%A3.md，不易于维护，可能导致其他平台跳转有问题。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#2283](https://gitcode.com/cann/ops-math/issues/2283) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> docs/zh/context所有md和对应的link ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4182	1 个月前
examples	[feat] Add aicpu op: ReduceSum Expand Reshape GreaterEqual Co-authored-by: FengHaozhan<fenghaozhan@huawei.com> # message auto-generated for no-merge-commit merge: !2829 merge Reshape-4-op into master [feat] Add aicpu op: ReduceSum Expand Reshape GreaterEqual Created-by: FengHaozhan Commit-by: FengHaozhan Merged-by: cann-robot Description: ## 描述本次改动为 Reshape、Expand、GreaterEqual、ReduceSum 四个算子补充了 AICPU 实现及相关支撑，具体包含： - Reshape: 新增 AICPU kernel, host infershape, example, UT, README/CMake - Expand: 新增 AICPU kernel, example, UT, README/CMake - GreaterEqual: 新增 AICPU kernel, example, UT, README/CMake - 更新了 docs/zh/op_list.md 文档，明确了相关算子的 AICPU 支持状态。 - 完善了 CMake 构建脚本，确保新增算子正确接入编译体系。 ## 关联的Issue #1621 ## 测试 - [ ] 编译构建验证：建议执行全量编译确保无冲突。 - [ ] AICPU 单元测试：已包含四个算子的 UT，建议在目标环境执行验证。 - [ ] Graph Example 基本验证：建议通过提供的 example 脚本跑通基本推断流程。 - [ ] 接口注册/加载验证：确认算子在系统启动时正确注册到 AICPU 算子库。 ## 文档更新 README 与 docs/zh/op_list.md 已按要求更新。 ## 类型标签 - [x] ✨ 新特性 - [ ] 🐛 修复 - [x] 🧪 测试 - [x] 📦 构建/CI - [x] 📝 文档更新 - [ ] 🎨 代码风格 - [ ] ♻️ 重构 - [ ] ⚡ 性能优化 - [ ] 🔒 安全修复 - [ ] 🏷️ 标签属性维护 - [ ] 🏷️ 废弃 See merge request: cann/ops-math!2829	3 个月前
op_api	Expand算子空tensor处理修改 Co-authored-by: l00939308<luwenxiang4@huawei.com> # message auto-generated for no-merge-commit merge: !1985 merge master into master Expand算子空tensor处理修改 Created-by: luwenxiang1998 Commit-by: l00939308 Merged-by: cann-robot Description: ## 描述 Expand算子空tensor处理修改 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/1119 <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!1985	5 个月前
op_graph	新增Expand算子 Co-authored-by: l00939308<luwenxiang4@huawei.com> # message auto-generated for no-merge-commit merge: !1133 merge master into master 新增Expand算子 Created-by: luwenxiang1998 Commit-by: l00939308 Merged-by: cann-robot Description: ## 描述新增Expand算子 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/814 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：新增Expand算子 See merge request: cann/ops-math!1133	6 个月前
op_host	cleancode fix Co-authored-by: yangjinwen<yangjinwen2@huawei.com> # message auto-generated for no-merge-commit merge: !3775 merge dev-cleancode into master cleancode fix Created-by: yangjinwen Commit-by: yangjinwen Merged-by: cann-robot Description: ## 描述 cleancode fix ## 关联的Issue https://gitcode.com/cann/ops-math/issues/1944 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3775	2 个月前
op_kernel	新增Expand算子 Co-authored-by: l00939308<luwenxiang4@huawei.com> # message auto-generated for no-merge-commit merge: !1133 merge master into master 新增Expand算子 Created-by: luwenxiang1998 Commit-by: l00939308 Merged-by: cann-robot Description: ## 描述新增Expand算子 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/814 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：新增Expand算子 See merge request: cann/ops-math!1133	6 个月前
op_kernel_aicpu	feat: 支持八个迁移算子的常量折叠(#2551) Co-authored-by: pan-tong<pantong@huawei.com> # message auto-generated for no-merge-commit merge: !4511 merge feat/constant-folding-infershape into master feat: 支持八个迁移算子的常量折叠(#2551) Created-by: pan-tong Commit-by: pan-tong Merged-by: cann-robot Description: ## 描述为 Ceil、ConcatV2、Expand、GreaterEqual、InvertPermutation、StridedSlice、StridedSliceV2、TransData 接入 HOSTCPU 常量折叠。 - 为尚未接线的 7 个算子在 CMake 中启用 `HOSTCPU TRUE`，将现有 AICPU kernel 编入 `libopconstant_folding_math.so`。 - 8 个算子的 AICPU kernel 使用 `OPS_MATH_REGISTER_CPU_KERNELV2`，兼容 device 注册和 host CPU Kernel V2 注册。 - 为 GreaterEqual 补齐 `OPS_FLAG_OPEN`；其余算子沿用已有 OPEN 配置。 - TransData 沿用已有 HOSTCPU 配置，并复用已接入 host object 的 `format_transfer` 辅助源。 - 增强 Ceil AICPU UT 的输出校验，新增 Ceil 全 Const GE 图常量折叠用例。本 PR 不修改算子的 infershape 实现。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2551 ## 测试 - AICPU UT：8 个算子共 156 个用例全部通过。 - HOSTCPU 构建：`libopconstant_folding_math.so` 构建成功，包含 8 个算子的 host 注册。 - 安装级 GE 常量折叠：8 个算子的全 Const 图输出均校验通过。 - 逐算子日志确认对应 host kernel 执行，`ConstantFoldingPass` 返回码均为 0。 - `pre-commit`：clang-format、codespell、OAT 等检查全部通过。 - GitCode 服务端 Hooks：通过。 ## 文档更新无。该改动不改变算子接口和使用方式。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4511	26 天前
tests	feat: 新增BroadcastTo等算子UT Co-authored-by: zhanw_coding<zhanwang@huawei.com> # message auto-generated for no-merge-commit merge: !3663 merge ft-ut into master feat: 新增BroadcastTo等算子UT Created-by: zhanw_coding Commit-by: zhanw_coding Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> feat: 新增BroadcastTo等算子UT 包括：broadcast_to,expand,tile,pad,pad_v3,mirror_pad,pad_v3_grad,circular_pad,circular_pad_grad,pad_v2,pad_v3_grad_replicate,pad_v3_grad_replication,reflection_pad3d_grad,transpose,depth_to_space,space_to_depth,batch_to_space_nd,sinc,truncate_div,truncate_mod,rint ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/2090 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：UT See merge request: cann/ops-math!3663	2 个月前
CMakeLists.txt	feat: 支持八个迁移算子的常量折叠(#2551) Co-authored-by: pan-tong<pantong@huawei.com> # message auto-generated for no-merge-commit merge: !4511 merge feat/constant-folding-infershape into master feat: 支持八个迁移算子的常量折叠(#2551) Created-by: pan-tong Commit-by: pan-tong Merged-by: cann-robot Description: ## 描述为 Ceil、ConcatV2、Expand、GreaterEqual、InvertPermutation、StridedSlice、StridedSliceV2、TransData 接入 HOSTCPU 常量折叠。 - 为尚未接线的 7 个算子在 CMake 中启用 `HOSTCPU TRUE`，将现有 AICPU kernel 编入 `libopconstant_folding_math.so`。 - 8 个算子的 AICPU kernel 使用 `OPS_MATH_REGISTER_CPU_KERNELV2`，兼容 device 注册和 host CPU Kernel V2 注册。 - 为 GreaterEqual 补齐 `OPS_FLAG_OPEN`；其余算子沿用已有 OPEN 配置。 - TransData 沿用已有 HOSTCPU 配置，并复用已接入 host object 的 `format_transfer` 辅助源。 - 增强 Ceil AICPU UT 的输出校验，新增 Ceil 全 Const GE 图常量折叠用例。本 PR 不修改算子的 infershape 实现。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2551 ## 测试 - AICPU UT：8 个算子共 156 个用例全部通过。 - HOSTCPU 构建：`libopconstant_folding_math.so` 构建成功，包含 8 个算子的 host 注册。 - 安装级 GE 常量折叠：8 个算子的全 Const 图输出均校验通过。 - 逐算子日志确认对应 host kernel 执行，`ConstantFoldingPass` 返回码均为 0。 - `pre-commit`：clang-format、codespell、OAT 等检查全部通过。 - GitCode 服务端 Hooks：通过。 ## 文档更新无。该改动不改变算子接口和使用方式。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4511	26 天前
README.md	批量修改md文档冗余空格第2批 Co-authored-by: chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !3314 merge master into master 批量修改md文档冗余空格第2批 Created-by: gitcode-chenjiao Commit-by: chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 批量修改md中冗余空格 ## 关联的Issue [DOC_1962#冗余空格](https://gitcode.com/cann/ops-math/issues/1962) ## 测试自验证 ## 文档更新 all md ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [X ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3314	2 个月前

Expand

产品支持情况

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

功能说明

算子功能：将输入tensor广播到指定的shape。如输入tensor的shape为(1, 4)，指定的shape为(2, 4)，则输出是shape为(2, 4)的tensor。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入张量	需要被广播的张量。	FLOAT16、FLOAT、INT32、INT64、INT8、UINT8、BOOL、BF16	ND
shape	输入张量	表示x广播后的shape大小。	INT64、INT32	-
y	输出	维度最大不超过8维，shape由shape输入决定，dtype需要与self一致。	同x	ND

约束说明

无

调用说明

调用方式	样例代码	说明
图模式调用	test_geir_expand	通过GE IR图模式调用Expand算子
aclnn接口	test_aclnn_expand	通过aclnnExpand接口方式调用Expand算子。