已合并
feat: Python 自定义算子原型执行样例 #4519
lfz2812创建于 14 天前
feat: Python 自定义算子原型执行样例 #4519
已合并
Pull Request已成功合入, 合并人@CANN-robot
(感谢 lfz2812 的贡献)14 天前 virtual merging failed, update merge request[project_id: 8824148, iid: 4519, target_commit_sha: 56f8e1bb90b1c89bcbbffe3dd952361bfa449121], message: Conflict detected
14 天前 virtual merging failed, update merge request[project_id: 8824148, iid: 4519, target_commit_sha: 56f8e1bb90b1c89bcbbffe3dd952361bfa449121], message: Conflict detected
此处折叠了572条消息 查看更多
yangyongqiang
7 天前 评论:
7 天前 评论:
/lgtm


7 天前 添加了label:lgtm
7 天前 添加了label:approved
7 天前 合入了pull request
Pull Request
描述
完善 Python 自定义算子公开资料,并将现有
args_refresh_add_custom/python样例升级为 Python 原型注册、infer-meta 和 PyPTO 执行的端到端样例。主要变更:
src/ge/add_custom.py中使用@register_op注册AddPythonCustomOp原型和 infer-meta。TensorDesc返回输出元信息。@register_op_impl注册 schema-boundexecute(x, y)实现。add_custom.asc等价的z = x + yNPU kernel。torch_npuTensor,并在 GE 当前 stream 上执行 PyPTO kernel。_load_kernel()、ACL binary 加载和 kernel launch 流程。run.sh,仅生成和安装 ES Python API,并运行 Python Session 样例。_out_shape_locked属性和手工设置输出 shape、format、data type 的逻辑,输出元信息由 Python infer-meta 提供。REG_OP仅作为gen_esb生成ge.es.custom.AddPythonCustomOp构造器的输入,不参与运行时原型注册。output0,与 Pythonregister_op对普通非别名输出生成的默认名称保持一致,避免 canonical IR 兼容性检查失败。ge.runtime.TensorDesc中文 API 文档,并说明其与ge.graph.TensorDesc的区别。变更类型
请选择本次引入的变更类型:
关联的Issue
#299
如何测试
已执行以下静态检查:
已针对全部变更文件执行 pre-commit,以下检查均通过:
NPU 环境验收步骤:
核对清单
其他信息