ops-math/math/right_shift/framework · yuwang-/ops-math

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

Files	Last commit	Last update
CMakeLists.txt	新增950算子 Co-authored-by: huangyuxiaaaaa<huangyuxia9@huawei.com> # message auto-generated for no-merge-commit merge: !1641 merge erf into master 新增950算子 Created-by: huangyuxiaaaaa Commit-by: huangyuxiaaaaa Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 新增950算子。涉及算子：atan、erf、expm1、nan_to_num、right_shift 涉及文件：tiling、kernel、infershape、ut、算子原型、算子信息库、算子相关配置 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#1035](https://gitcode.com/cann/ops-math/issues/1035) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 已通过本地验证、静态检查、蓝区冒烟、OBP和David冒烟 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：算子开源 See merge request: cann/ops-math!1641	7 months ago
bitshift_onnx_plugin.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	16 days ago