| Files | Last commit | Last update |
|---|---|---|
feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !6621 merge migrate-gelocal-delivery into master feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: ## 描述 汇总迁移 gelocal（RT2.0）算子交付件： - **基础/GE 控制流算子**：size、rank、LoopCond、NextIteration、RefIdentity、SwitchByIndex、aipp_data、End、squeezev2、unsqueezev2，以及 enter/exit/merge/switch/switch_n/ref_* 等 - **子图控制流算子**：If、Case、For、PartitionedCall、While（原型注册 + RT2.0 推导注册 + UT + README，If/Case/For/PartitionedCall 含 GeIR 示例；均归入 ge 分类） - **TF 控制流插件**：If/_If/StatelessIf、While/_While/StatelessWhile、Case/StatelessCase、PartitionedCall/StatefulPartitionedCall 变体合并映射 - **reshape**：infershape 对齐 RT1.0 语义、allowzero/shape 属性注册、信息库 - **文档**：README 产品支持表统一（全部√）、gather_shapes 归入 conversion、op_list 同步登记 ## 测试 - bash build.sh --ophost --opgraph -j8 构建通过（含各算子 REG_OP 注册校验） - 宿主侧全量 UT：2594/2594 PASSED（384 test suites） ## 说明 - 内容汇总自此前迁移 PR（6474 / 6522 / 6208）的交付件，本 PR 为其合并后的统一交付 See merge request: cann/ops-math!6621 | 16 days ago | |
math类onnx算子插件支持 Co-authored-by: x00842564<xuyanke@huawei.com> Co-authored-by: weiyukun<weiyukun2@h-partners.com> # message auto-generated for no-merge-commit merge: !431 merge master into master math类onnx算子插件支持 Created-by: yanke-xu Commit-by: weiyukun;x00842564 Merged-by: cann-robot Description: ## 描述 onnx算子支持列表： 1. addcmul 2. ArgMax 3. ArgMin 4. Bernoulli 5. BitShift 6. Cast 7. Clip 8. Concat 9. ConcatFromSequence 10. Corr 11. CumSum 12. DeformableOffsets 13. DepthToSpace 14. Einsum 15. EyeLike 16. Flatten 17. IsInf 18. Max 19. Mean 20. Min 21. Mod 22. Multinomial 23. NPUDtypeCast 24. NPUFormatCast 25. NPUOneHot 26. NPURotaryMul 27. NPUSignBitsPack 28. NPUSignBitsUnpack 29. NPUSlice 30. OneHot 31. Pad 32. PhonyConcat 33. PhonySplit 34. RandomNormal 35. RandomNormalLike 36. RandomUniform 37. RandomUniformLike 38. ReduceL1 39. ReduceL2 40. ReduceLogSum 41. ReduceLogSumExp 42. ReduceMax 43. ReduceMean 44. ReduceMin 45. ReduceProd 46. ReduceSum 47. ReduceSumSquare 48. RepeatWithPad 49. Reshape 50. ReverseSequence 51. SequenceAt 52. SequenceConstruct 53. SequenceEmpty 54. SequenceErase 55. SequenceInsert 56. SequenceLength 57. Slice 58. SpaceToDepth 59. Split 60. SplitToSequence 61. Squeeze 62. Sum 63. Transpose 64. Trilu 65. Unsqueeze 66. Xor ## 关联的Issue None ## 测试 1、编译构建 2、静态检查 3、冒烟测试 4、本地功能验证 ## 文档更新 None ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!431 | 10 months ago | |
[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 | 5 months ago | |
feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !6621 merge migrate-gelocal-delivery into master feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: ## 描述 汇总迁移 gelocal（RT2.0）算子交付件： - **基础/GE 控制流算子**：size、rank、LoopCond、NextIteration、RefIdentity、SwitchByIndex、aipp_data、End、squeezev2、unsqueezev2，以及 enter/exit/merge/switch/switch_n/ref_* 等 - **子图控制流算子**：If、Case、For、PartitionedCall、While（原型注册 + RT2.0 推导注册 + UT + README，If/Case/For/PartitionedCall 含 GeIR 示例；均归入 ge 分类） - **TF 控制流插件**：If/_If/StatelessIf、While/_While/StatelessWhile、Case/StatelessCase、PartitionedCall/StatefulPartitionedCall 变体合并映射 - **reshape**：infershape 对齐 RT1.0 语义、allowzero/shape 属性注册、信息库 - **文档**：README 产品支持表统一（全部√）、gather_shapes 归入 conversion、op_list 同步登记 ## 测试 - bash build.sh --ophost --opgraph -j8 构建通过（含各算子 REG_OP 注册校验） - 宿主侧全量 UT：2594/2594 PASSED（384 test suites） ## 说明 - 内容汇总自此前迁移 PR（6474 / 6522 / 6208）的交付件，本 PR 为其合并后的统一交付 See merge request: cann/ops-math!6621 | 16 days ago | |
feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !6621 merge migrate-gelocal-delivery into master feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: ## 描述 汇总迁移 gelocal（RT2.0）算子交付件： - **基础/GE 控制流算子**：size、rank、LoopCond、NextIteration、RefIdentity、SwitchByIndex、aipp_data、End、squeezev2、unsqueezev2，以及 enter/exit/merge/switch/switch_n/ref_* 等 - **子图控制流算子**：If、Case、For、PartitionedCall、While（原型注册 + RT2.0 推导注册 + UT + README，If/Case/For/PartitionedCall 含 GeIR 示例；均归入 ge 分类） - **TF 控制流插件**：If/_If/StatelessIf、While/_While/StatelessWhile、Case/StatelessCase、PartitionedCall/StatefulPartitionedCall 变体合并映射 - **reshape**：infershape 对齐 RT1.0 语义、allowzero/shape 属性注册、信息库 - **文档**：README 产品支持表统一（全部√）、gather_shapes 归入 conversion、op_list 同步登记 ## 测试 - bash build.sh --ophost --opgraph -j8 构建通过（含各算子 REG_OP 注册校验） - 宿主侧全量 UT：2594/2594 PASSED（384 test suites） ## 说明 - 内容汇总自此前迁移 PR（6474 / 6522 / 6208）的交付件，本 PR 为其合并后的统一交付 See merge request: cann/ops-math!6621 | 16 days ago | |
/approve Co-authored-by: ayuemoon<15802951036@163.com> # message auto-generated for no-merge-commit merge: !5334 merge aicpu-const-folding-reshape into master feat: enable aicpu constant folding for reshape Created-by: ayuemoon Commit-by: ayuemoon Merged-by: cann-robot Description: ## 描述 为 Reshape AICPU 算子接入 GE 图编译期常量折叠支持。 实现内容： - 在 conversion/reshape/CMakeLists.txt 中增加 HOSTCPU TRUE，将 Reshape kernel 编译为 host 侧常量折叠 OBJECT。 - 将 Reshape kernel 从 REGISTER_CPU_KERNEL 切换为 OPS_MATH_REGISTER_CPU_KERNELV2，host 编译时注册到 V2 kernel 表，device 编译行为保持不变。 - Reshape 算子已有 opsFlag=OPEN，本次不修改算子计算逻辑。 ## 关联的Issue 无。 ## 测试 已完成： - git diff --check。 - 确认 opsFlag=OPEN、HOSTCPU TRUE 和 V2 kernel 注册均生效。 - Reshape AICPU kernel 未发现 ascend_hal、halSdma、SetCopyHook 等 host 不可用依赖。 - Git push hook 检查通过。 当前环境未配置 CANN 工具链，未执行实际 host so 编译、run 包打包及 GE 端到端常量折叠验证。 ## 文档更新 无。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [x] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5334 | 2 months ago | |
feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !6621 merge migrate-gelocal-delivery into master feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: ## 描述 汇总迁移 gelocal（RT2.0）算子交付件： - **基础/GE 控制流算子**：size、rank、LoopCond、NextIteration、RefIdentity、SwitchByIndex、aipp_data、End、squeezev2、unsqueezev2，以及 enter/exit/merge/switch/switch_n/ref_* 等 - **子图控制流算子**：If、Case、For、PartitionedCall、While（原型注册 + RT2.0 推导注册 + UT + README，If/Case/For/PartitionedCall 含 GeIR 示例；均归入 ge 分类） - **TF 控制流插件**：If/_If/StatelessIf、While/_While/StatelessWhile、Case/StatelessCase、PartitionedCall/StatefulPartitionedCall 变体合并映射 - **reshape**：infershape 对齐 RT1.0 语义、allowzero/shape 属性注册、信息库 - **文档**：README 产品支持表统一（全部√）、gather_shapes 归入 conversion、op_list 同步登记 ## 测试 - bash build.sh --ophost --opgraph -j8 构建通过（含各算子 REG_OP 注册校验） - 宿主侧全量 UT：2594/2594 PASSED（384 test suites） ## 说明 - 内容汇总自此前迁移 PR（6474 / 6522 / 6208）的交付件，本 PR 为其合并后的统一交付 See merge request: cann/ops-math!6621 | 16 days ago | |
/approve Co-authored-by: ayuemoon<15802951036@163.com> # message auto-generated for no-merge-commit merge: !5334 merge aicpu-const-folding-reshape into master feat: enable aicpu constant folding for reshape Created-by: ayuemoon Commit-by: ayuemoon Merged-by: cann-robot Description: ## 描述 为 Reshape AICPU 算子接入 GE 图编译期常量折叠支持。 实现内容： - 在 conversion/reshape/CMakeLists.txt 中增加 HOSTCPU TRUE，将 Reshape kernel 编译为 host 侧常量折叠 OBJECT。 - 将 Reshape kernel 从 REGISTER_CPU_KERNEL 切换为 OPS_MATH_REGISTER_CPU_KERNELV2，host 编译时注册到 V2 kernel 表，device 编译行为保持不变。 - Reshape 算子已有 opsFlag=OPEN，本次不修改算子计算逻辑。 ## 关联的Issue 无。 ## 测试 已完成： - git diff --check。 - 确认 opsFlag=OPEN、HOSTCPU TRUE 和 V2 kernel 注册均生效。 - Reshape AICPU kernel 未发现 ascend_hal、halSdma、SetCopyHook 等 host 不可用依赖。 - Git push hook 检查通过。 当前环境未配置 CANN 工具链，未执行实际 host so 编译、run 包打包及 GE 端到端常量折叠验证。 ## 文档更新 无。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [x] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5334 | 2 months ago | |
feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !6621 merge migrate-gelocal-delivery into master feat: 迁移gelocal控制流与基础算子交付件（含While/If/Case/For/PartitionedCall、TF控制流插件、reshape），实现RT2.0交付 Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: ## 描述 汇总迁移 gelocal（RT2.0）算子交付件： - **基础/GE 控制流算子**：size、rank、LoopCond、NextIteration、RefIdentity、SwitchByIndex、aipp_data、End、squeezev2、unsqueezev2，以及 enter/exit/merge/switch/switch_n/ref_* 等 - **子图控制流算子**：If、Case、For、PartitionedCall、While（原型注册 + RT2.0 推导注册 + UT + README，If/Case/For/PartitionedCall 含 GeIR 示例；均归入 ge 分类） - **TF 控制流插件**：If/_If/StatelessIf、While/_While/StatelessWhile、Case/StatelessCase、PartitionedCall/StatefulPartitionedCall 变体合并映射 - **reshape**：infershape 对齐 RT1.0 语义、allowzero/shape 属性注册、信息库 - **文档**：README 产品支持表统一（全部√）、gather_shapes 归入 conversion、op_list 同步登记 ## 测试 - bash build.sh --ophost --opgraph -j8 构建通过（含各算子 REG_OP 注册校验） - 宿主侧全量 UT：2594/2594 PASSED（384 test suites） ## 说明 - 内容汇总自此前迁移 PR（6474 / 6522 / 6208）的交付件，本 PR 为其合并后的统一交付 See merge request: cann/ops-math!6621 | 16 days ago |
Reshape
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR&950DT系列产品 | √ |
| Atlas A3系列产品 | √ |
| Atlas A2系列产品 | √ |
| Atlas 200I/500 A2推理产品 | √ |
| Atlas推理系列产品 | √ |
| Atlas训练系列产品 | √ |
功能说明
- 算子功能：在不改变底层数据布局的前提下，将输入张量重解释为目标形状。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x | 输入 | 待重塑的输入张量。 | BOOL, FLOAT16, FLOAT, INT4, INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, DOUBLE, COMPLEX64 | ND、FRACTAL_NZ |
| shape | 输入 | 目标形状张量，支持一个-1；allowzero为0时0表示复制对应输入维度。 |
INT32, INT64 | 1D |
| axis | 属性 | 起始替换维度，默认值为0。 | INT | - |
| num_axes | 属性 | 被替换的连续维度数，默认值为-1。 | INT | - |
| y | 输出 | 重塑后的输出张量，数据类型与输入x相同。 | 与x相同 | ND、FRACTAL_NZ |
约束说明
- 输出元素总数必须与输入元素总数一致。
- shape中最多只能出现一个
-1。 - 当allowzero为0时，shape中的
0会复制对应输入维度；当allowzero未设置或为非0值时，0按字面值参与计算。
调用说明
| 调用方式 | 样例代码 | 说明 |
|---|---|---|
| 图模式调用 | test_geir_reshape | 通过 算子IR构图方式调用reshape算子。 |