提交 77ead937 编写于 作者: M Megvii Engine Team 提交者: huangxinda

fix(src/serialization): fix compatibility error of oss model

GitOrigin-RevId: 43e0fa4fe14cbfdcbc835a35487368098e9772ed
上级 070c8117
...@@ -309,7 +309,7 @@ def optimize_for_inference(args, outputs): ...@@ -309,7 +309,7 @@ def optimize_for_inference(args, outputs):
"enable_chwn4", "enable_chwn4",
"enable_fuse_conv_bias_nonlinearity", "enable_fuse_conv_bias_nonlinearity",
"enable_fuse_conv_bias_with_z", "enable_fuse_conv_bias_with_z",
"eaable_fuse_preprocess", "enable_fuse_preprocess",
] ]
kwargs = {} kwargs = {}
for k in args_list: for k in args_list:
......
...@@ -723,15 +723,17 @@ GraphLoaderOSS::OprLoadContextImpl::load_tensor_shared() { ...@@ -723,15 +723,17 @@ GraphLoaderOSS::OprLoadContextImpl::load_tensor_shared() {
Metadata GraphLoaderOSS::OprLoadContextImpl::load_metadata() { Metadata GraphLoaderOSS::OprLoadContextImpl::load_metadata() {
const auto* fbmeta = m_loader->m_graph->metadata(); const auto* fbmeta = m_loader->m_graph->metadata();
Metadata ret; Metadata ret;
ret.is_valid = fbmeta->is_valid(); if (fbmeta) {
ret.graph_modified = fbmeta->graph_modified(); ret.is_valid = fbmeta->is_valid();
if (fbmeta->user_info()) { ret.graph_modified = fbmeta->graph_modified();
ret.user_info = fbmeta->user_info()->str(); if (fbmeta->user_info()) {
ret.has_user_info = true; ret.user_info = fbmeta->user_info()->str();
} ret.has_user_info = true;
if (fbmeta->optimize_options()) { }
ret.optimize_options = fbmeta->optimize_options(); if (fbmeta->optimize_options()) {
ret.optimized_for_inference = true; ret.optimize_options = fbmeta->optimize_options();
ret.optimized_for_inference = true;
}
} }
return ret; return ret;
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册