From a4f0e581802f0b1fc6782e967f6b70bb2416d582 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 24 Aug 2021 18:51:57 +0800 Subject: [PATCH] fix(mgb/extern_c_opr): throw exception when extern c opr loader was created GitOrigin-RevId: bf48e1f3bb9fce733174522024504e9b1289c5be --- src/serialization/impl/extern_c_opr.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/serialization/impl/extern_c_opr.cpp b/src/serialization/impl/extern_c_opr.cpp index a8b42cdb..c6cc3cfb 100644 --- a/src/serialization/impl/extern_c_opr.cpp +++ b/src/serialization/impl/extern_c_opr.cpp @@ -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); } -- GitLab