基于 upstream/develop 实现 Python 化可编译自定义算子:Python 自定义算子可以声明 compile 接口,并在在线图编译阶段通过真实 OpCompileContext 获取编译 option、平台资源、核数和 SoC 信息。
upstream/develop
compile
OpCompileContext
主要变更:
compilable
get_compile_ctx()
无。
ge_compiler
atc_static
ge_executor
ge_python_custom_op_bridge
custom_op_runtime
CustomGraphOptimizer::OptimizeWholeGraph
feat:
fix:
cann/ge:develop
b300e9b3cda3a4563a7a1d6b922f5df8929f309d
develop
/lgtm
/approve
描述
基于
upstream/develop实现 Python 化可编译自定义算子:Python 自定义算子可以声明compile接口,并在在线图编译阶段通过真实OpCompileContext获取编译 option、平台资源、核数和 SoC 信息。主要变更:
compilable能力收集、schema 参数绑定、get_compile_ctx()及 nativeOpCompileContext绑定。compile回调。变更类型
关联的Issue
无。
如何测试
ge_compiler、atc_static、ge_executor、ge_python_custom_op_bridge、custom_op_runtime和相关测试目标成功;完整目标未宣称全量构建通过。CustomGraphOptimizer::OptimizeWholeGraph触发真实 Python compile 回调及离线跳过门禁的 ST:2 passed。核对清单
其他信息
cann/ge:develop,基线提交b300e9b3cda3a4563a7a1d6b922f5df8929f309d。