| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
reshape算子补齐rt2的infershape Co-authored-by: zcy123456x<zhouchenyang5@huawei.com> Co-authored-by: zcy123456x<zcy123456x@noreply.gitcode.com> # message auto-generated for no-merge-commit merge: !5439 merge master into master reshape算子补齐rt2的infershape Created-by: zcy123456x Commit-by: zcy123456x Merged-by: cann-robot Description: ## 描述 本 PR 为 Reshape 算子补齐 rt2 的推理实现,新增了 op_host 侧的 InferShape(reshape_infershape.cpp)与 op_graph 侧的数据类型推理(reshape_graph_infer.cpp)逻辑,并配套新增两组单元测试,用于校验输出形状与输出数据类型推导的正确性。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!5439 | 11 天前 | |
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 | 8 个月前 | |
[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 | 4 个月前 | |
reshape算子补齐rt2的infershape Co-authored-by: zcy123456x<zhouchenyang5@huawei.com> Co-authored-by: zcy123456x<zcy123456x@noreply.gitcode.com> # message auto-generated for no-merge-commit merge: !5439 merge master into master reshape算子补齐rt2的infershape Created-by: zcy123456x Commit-by: zcy123456x Merged-by: cann-robot Description: ## 描述 本 PR 为 Reshape 算子补齐 rt2 的推理实现,新增了 op_host 侧的 InferShape(reshape_infershape.cpp)与 op_graph 侧的数据类型推理(reshape_graph_infer.cpp)逻辑,并配套新增两组单元测试,用于校验输出形状与输出数据类型推导的正确性。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!5439 | 11 天前 | |
reshape算子补齐rt2的infershape Co-authored-by: zcy123456x<zhouchenyang5@huawei.com> Co-authored-by: zcy123456x<zcy123456x@noreply.gitcode.com> # message auto-generated for no-merge-commit merge: !5439 merge master into master reshape算子补齐rt2的infershape Created-by: zcy123456x Commit-by: zcy123456x Merged-by: cann-robot Description: ## 描述 本 PR 为 Reshape 算子补齐 rt2 的推理实现,新增了 op_host 侧的 InferShape(reshape_infershape.cpp)与 op_graph 侧的数据类型推理(reshape_graph_infer.cpp)逻辑,并配套新增两组单元测试,用于校验输出形状与输出数据类型推导的正确性。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!5439 | 11 天前 | |
/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 | 22 天前 | |
reshape算子补齐rt2的infershape Co-authored-by: zcy123456x<zhouchenyang5@huawei.com> Co-authored-by: zcy123456x<zcy123456x@noreply.gitcode.com> # message auto-generated for no-merge-commit merge: !5439 merge master into master reshape算子补齐rt2的infershape Created-by: zcy123456x Commit-by: zcy123456x Merged-by: cann-robot Description: ## 描述 本 PR 为 Reshape 算子补齐 rt2 的推理实现,新增了 op_host 侧的 InferShape(reshape_infershape.cpp)与 op_graph 侧的数据类型推理(reshape_graph_infer.cpp)逻辑,并配套新增两组单元测试,用于校验输出形状与输出数据类型推导的正确性。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!5439 | 11 天前 | |
/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 | 22 天前 | |
产品名变更 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !5912 merge master into master 产品名变更 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> AI处理器对应的产品型号名称更新 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> [#3258](https://gitcode.com/cann/ops-math/issues/3258) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!5912 | 7 天前 |
Reshape
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR&950DT系列产品 | √ |
| Atlas A3系列产品 | √ |
| Atlas A2系列产品 | √ |
| Atlas 200I/500 A2推理产品 | × |
| Atlas推理系列产品 | √ |
| Atlas训练系列产品 | √ |
功能说明
- 算子功能:在不改变底层数据布局的前提下,将输入张量重解释为目标形状。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x | 输入 | 待重塑的输入张量。 | BOOL, FLOAT, FLOAT16, INT8, INT16, UINT16, UINT8, INT32, INT64, UINT32, UINT64, DOUBLE, COMPLEX64, COMPLEX128, BF16 | ND |
| shape | 输入 | 目标形状张量,支持一个 -1;默认模式下 0 表示复制对应输入维度。 |
INT32, INT64 | 1D |
| axis | 属性 | 起始替换维度,默认值为0。 | INT | - |
| num_axes | 属性 | 被替换的连续维度数,默认值为 -1。 | INT | - |
| y | 输出 | 重塑后的输出张量,数据类型与输入x相同。 | 与x相同 | ND |
约束说明
- 输出元素总数必须与输入元素总数一致。
- shape中最多只能出现一个
-1。 - 当allowzero未设置或为0时,shape中的
0会复制对应输入维度;当allowzero为1时,0按字面值参与计算。
调用说明
| 调用方式 | 样例代码 | 说明 |
|---|---|---|
| 图模式调用 | test_geir_reshape | 通过 算子IR构图方式调用reshape算子。 |