ops-math/math/left_shift/op_graph · SeanDictionary/ops-math

cann-robot[LeftShift] 新增 Ascend 950 / arch35 Ascend C 实现

Files	Last commit	Last update
left_shift_graph_infer.cpp	[LeftShift] 新增 Ascend 950 / arch35 Ascend C 实现 Co-authored-by: gl_princekin<gelei10@huawei.com> # message auto-generated for no-merge-commit merge: !6684 merge master into master [LeftShift] 新增 Ascend 950 / arch35 Ascend C 实现 Created-by: gl_princekin Commit-by: gl_princekin Merged-by: cann-robot Description: ## 描述 LeftShift（按位左移）算子此前仅包含 aclnn 接口（AICPU 实现），缺少 Ascend C 原生实现。本 PR 新增 LeftShift 的 Ascend C 实现，使其在 Ascend 950 及 Atlas A3 系列（arch35 架构）产品上获得原生执行路径。主要改动： - `op_kernel/arch35/`：新增 Ascend C kernel（DAG 节点 + 数据结构定义） - `op_host/arch35/`：新增 tiling 代码 - `op_host/config/ascend950/`、`op_host/config/ascend350/`：算子配置（JSON + simplified_key.ini） - `op_host/`：`left_shift_def.cpp`（算子定义）、`left_shift_infershape.cpp`（shape 推导） - `op_graph/`：`left_shift_proto.h`、`left_shift_graph_infer.cpp` - `op_kernel_aicpu/`：AICPU kernel 实现 - `op_api/`：aclnn 接口参数校验代码更新 - `examples/`：图模式调用示例 `test_geir_left_shift.cpp` - `tests/`：单元测试（op_api / op_host / tiling / infershape / ST） - `README.md`、`docs/`：产品支持表、广播说明文档 - `common/inc/op_graph/op_math_proto_extend.h`：移除 LeftShift 注册（迁移至独立 proto 文件） - `math/right_shift/framework/bitshift_onnx_plugin.cpp`：修复 proto 头文件引用 ## 关联的Issue - #3978 ## 测试 - op_api 单元测试：`test_aclnn_left_shift.cpp`、`test_aclnn_left_shift_l0.cpp`、`test_aclnn_left_shifts.cpp` - op_host 单元测试：`test_left_shift_tiling.cpp`（arch35）、`test_left_shift_infershape.cpp` - op_kernel_aicpu 单元测试：`test_left_shift.cpp` - ST 测试：`tests/st/arch35/ttk_kernel_left_shift_st.csv` ## 文档更新 - 更新了 `math/left_shift/README.md`（产品支持表、功能说明、参数说明、约束说明、调用说明） - 更新了 `math/left_shift/docs/aclnnLeftShift.md`、`aclnnLeftShifts.md`（新增广播说明） ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6684	14 days ago
left_shift_proto.h	[LeftShift] 新增 Ascend 950 / arch35 Ascend C 实现 Co-authored-by: gl_princekin<gelei10@huawei.com> # message auto-generated for no-merge-commit merge: !6684 merge master into master [LeftShift] 新增 Ascend 950 / arch35 Ascend C 实现 Created-by: gl_princekin Commit-by: gl_princekin Merged-by: cann-robot Description: ## 描述 LeftShift（按位左移）算子此前仅包含 aclnn 接口（AICPU 实现），缺少 Ascend C 原生实现。本 PR 新增 LeftShift 的 Ascend C 实现，使其在 Ascend 950 及 Atlas A3 系列（arch35 架构）产品上获得原生执行路径。主要改动： - `op_kernel/arch35/`：新增 Ascend C kernel（DAG 节点 + 数据结构定义） - `op_host/arch35/`：新增 tiling 代码 - `op_host/config/ascend950/`、`op_host/config/ascend350/`：算子配置（JSON + simplified_key.ini） - `op_host/`：`left_shift_def.cpp`（算子定义）、`left_shift_infershape.cpp`（shape 推导） - `op_graph/`：`left_shift_proto.h`、`left_shift_graph_infer.cpp` - `op_kernel_aicpu/`：AICPU kernel 实现 - `op_api/`：aclnn 接口参数校验代码更新 - `examples/`：图模式调用示例 `test_geir_left_shift.cpp` - `tests/`：单元测试（op_api / op_host / tiling / infershape / ST） - `README.md`、`docs/`：产品支持表、广播说明文档 - `common/inc/op_graph/op_math_proto_extend.h`：移除 LeftShift 注册（迁移至独立 proto 文件） - `math/right_shift/framework/bitshift_onnx_plugin.cpp`：修复 proto 头文件引用 ## 关联的Issue - #3978 ## 测试 - op_api 单元测试：`test_aclnn_left_shift.cpp`、`test_aclnn_left_shift_l0.cpp`、`test_aclnn_left_shifts.cpp` - op_host 单元测试：`test_left_shift_tiling.cpp`（arch35）、`test_left_shift_infershape.cpp` - op_kernel_aicpu 单元测试：`test_left_shift.cpp` - ST 测试：`tests/st/arch35/ttk_kernel_left_shift_st.csv` ## 文档更新 - 更新了 `math/left_shift/README.md`（产品支持表、功能说明、参数说明、约束说明、调用说明） - 更新了 `math/left_shift/docs/aclnnLeftShift.md`、`aclnnLeftShifts.md`（新增广播说明） ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6684	14 days ago