Browse Source

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

GitOrigin-RevId: 43e0fa4fe1
release-1.5
Megvii Engine Team huangxinda 3 years ago
parent
commit
77ead9377b
2 changed files with 12 additions and 10 deletions
  1. +1
    -1
      sdk/load-and-run/dump_with_testcase_mge.py
  2. +11
    -9
      src/serialization/impl/serializer_oss.cpp

+ 1
- 1
sdk/load-and-run/dump_with_testcase_mge.py View File

@@ -309,7 +309,7 @@ def optimize_for_inference(args, outputs):
"enable_chwn4",
"enable_fuse_conv_bias_nonlinearity",
"enable_fuse_conv_bias_with_z",
"eaable_fuse_preprocess",
"enable_fuse_preprocess",
]
kwargs = {}
for k in args_list:


+ 11
- 9
src/serialization/impl/serializer_oss.cpp View File

@@ -723,15 +723,17 @@ GraphLoaderOSS::OprLoadContextImpl::load_tensor_shared() {
Metadata GraphLoaderOSS::OprLoadContextImpl::load_metadata() {
const auto* fbmeta = m_loader->m_graph->metadata();
Metadata ret;
ret.is_valid = fbmeta->is_valid();
ret.graph_modified = fbmeta->graph_modified();
if (fbmeta->user_info()) {
ret.user_info = fbmeta->user_info()->str();
ret.has_user_info = true;
}
if (fbmeta->optimize_options()) {
ret.optimize_options = fbmeta->optimize_options();
ret.optimized_for_inference = true;
if (fbmeta) {
ret.is_valid = fbmeta->is_valid();
ret.graph_modified = fbmeta->graph_modified();
if (fbmeta->user_info()) {
ret.user_info = fbmeta->user_info()->str();
ret.has_user_info = true;
}
if (fbmeta->optimize_options()) {
ret.optimize_options = fbmeta->optimize_options();
ret.optimized_for_inference = true;
}
}
return ret;
}


Loading…
Cancel
Save