Browse Source

fix(mgb/extern_c_opr): throw exception when extern c opr loader was created

GitOrigin-RevId: bf48e1f3bb
tags/v1.6.0
Megvii Engine Team 4 years ago
parent
commit
a4f0e58180
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      src/serialization/impl/extern_c_opr.cpp

+ 4
- 0
src/serialization/impl/extern_c_opr.cpp View File

@@ -515,6 +515,10 @@ cg::OperatorNodeBase* ExternCOprRunner::load(OprLoadContext& ctx,
auto data = ctx.load_shared_buf_with_len();
auto desc = iter->second.first.create_desc(inputs.size(), data.data(),
data.size());

mgb_throw_if(nullptr == desc, MegBrainError,
"loader create desc returns nullptr");

if (auto trans = iter->second.second) {
desc = trans(desc);
}


Loading…
Cancel
Save