已合并
feat: 支持 Python 自定义算子原型与实现桥接注册 #4320
lfz2812创建于 11 天前
feat: 支持 Python 自定义算子原型与实现桥接注册 #4320
已合并
Pull Request已成功合入, 合并人@CANN-robot
(感谢 lfz2812 的贡献)此处折叠了518条消息 查看更多
黄桂军
3 天前 评论:
3 天前 评论:
/lgtm


3 天前 添加了label:lgtm
3 天前 添加了label:approved
3 天前 合入了pull request
Pull Request
描述
支持 Python 自定义算子原型从 Python descriptor 到 GE OperatorFactory 的注册,并完善 Python 自定义算子实现的 bridge 注册、校验和生命周期管理。
主要变更:
mutates_args,生成并深拷贝 Python custom op proto。OperatorFactorycreator,并通过公开 IR 查询接口获取 canonical IR。infer_meta,暂不覆盖编译期 Meta 推导和 RT2 runtime infer。变更类型
请选择本次引入的变更类型:
关联的Issue
#299
如何测试
本次补充和扩展以下测试:
Python UT:
execute签名校验。C++ UT:
ST:
建议执行:
核对清单
其他信息