已合并
新增 TF plugin 编译机制 #1209
zhaowenrui创建于 20 天前
新增 TF plugin 编译机制 #1209
已合并
Pull Request已成功合入, 合并人@CANN-robot
(感谢 zhaowenrui 的贡献)20 天前 创建了 pull request,commit cee0e332
20 天前 修改了pull request 的描述
20 天前 virtual merging failed, update merge request[project_id: 7657293, iid: 1209, target_commit_sha: cee0e33217335133b6fbac2ebf224d83d84c2b27], message: Conflict detected
20 天前 virtual merging failed, update merge request[project_id: 7657293, iid: 1209, target_commit_sha: cee0e33217335133b6fbac2ebf224d83d84c2b27], message: Conflict detected
20 天前 强制推送 1 个提交:92078e29-feat: add tf plugin build mechanism
20 天前 virtual merging failed, update merge request[project_id: 7657293, iid: 1209, target_commit_sha: cee0e33217335133b6fbac2ebf224d83d84c2b27], message: Conflict detected
20 天前 virtual merging failed, update merge request[project_id: 7657293, iid: 1209, target_commit_sha: cee0e33217335133b6fbac2ebf224d83d84c2b27], message: Conflict detected
此处折叠了67条消息 查看更多
14 天前 添加了label:approved
TangPC
14 天前 评论:
14 天前 评论:
/lgtm
/approve


14 天前 添加了label:lgtm
14 天前 关闭了关联的issue
14 天前 合入了pull request
描述
新增 ops-cv TensorFlow plugin 编译机制:
optf_plugin_cvtarget 及--tfplugin独立构建入口。--pkg内置包构建额外携带liboptf_plugin_cv.so,对齐 ops-math/ops-nn 的 TF plugin 包装方式。ge_ir.protoheader 初始化,并通过common/src/framework/cv_stub_tf_plugin.cpp保持目标可构建。add_tf_plugin_sources()作为单算子目录显式接入入口。framework/*_tf_plugin.cpp,因为 cv 仓现有部分 TF plugin 文件依赖非公开 GE 头文件;后续文件整改后可打开ENABLE_AUTO_TF_PLUGIN_SOURCES。关联的Issue
#706
测试
已完成:
bash -n build.sh scripts/build_conf.sh scripts/build_options.sh scripts/build_lib.shbash build.sh --tfplugin --ccache=offBuilt target optf_plugin_cv_objBuilt target optf_plugin_cvbuild/liboptf_plugin_cv.sobuild/CMakeFiles/optf_plugin_cv_obj.dir/common/src/framework/cv_stub_tf_plugin.cpp.o/home/developer/.local/bin/pre-commit run --files <本 PR 修改文件>回归结论:
--tfplugin独立构建通过。ENABLE_JIT未参与 TF plugin 库目标选择,未改变现有--jit语义。ENABLE_TEST=TRUE、非 installed CANN package 场景下 TF plugin CMake 入口直接返回,不引入 UT 路径。注意事项:
*_tf_plugin.cpp若仍依赖非公开 GE 头文件,需要由对应组件后续整改后再纳入自动收集。文档更新
无。
类型标签