Method load

src/serialization/impl/extern_c_opr.cpp:494–553

}

cg::OperatorNodeBase* ExternCOprRunner::load(
        OprLoadContext& ctx, const cg::VarNodeArray& inputs,
        const OperatorNodeConfig& config) {
    auto dump_name = ctx.load_buf_with_len();
    auto name = dump_name;
    //! use to compat dump ExternCOprRunner with more info
    if (auto index = name.find(":"))
        name = name.substr(0, index);
    auto&& map = loader_map();
    auto iter = map.find(name);
    // !!For MegCC loader.
    // If MegCC compiles a model containing an extern opr, it needs to use an extern
    // opr loader. In order to make MegCC's compiler not dependent on the extern opr
    // loader, MegCC registers a dummy loader with the name "extern_opr_dummy".
    bool dummy_loader = false;
    if (iter == map.end()) {
        mgb_log_debug(
                "Can NOT find loader '%s', try to find the dummy loader '%s'\n",
                name.c_str(), DUMMY_LOADER_NAME);
        iter = map.find(DUMMY_LOADER_NAME);
        if (iter != map.end()) {
            dummy_loader = true;
            mgb_log_debug("Found the dummy extern opr loader.\n");
        } else {
            mgb_log_debug("Can NOT find the dummy loader '%s'\n", DUMMY_LOADER_NAME);
        }
    }
    mgb_assert(
            iter != map.end(), "can not find loader for ExternCOprRunner '%s'",
            name.c_str());
    auto data = ctx.load_shared_buf_with_len();
    MGBOprDesc* desc = nullptr;

    // For MegCC loader.
    // If the loader is a MegCC dummy loader, copy the 'dump_name' in front of the
    // 'data' to relate 'data' to 'name'. So multi loaders and multi extern oprs in
    // single loader are supported.
    if (dummy_loader) {
        size_t buf_len = data.size() + sizeof(size_t) + dump_name.size();
        std::shared_ptr<void> buf{malloc(buf_len), free};
        char* buf_ptr = reinterpret_cast<char*>(buf.get());
        *(size_t*)buf_ptr = dump_name.size();
        buf_ptr += sizeof(size_t);
        memmove(buf_ptr, dump_name.c_str(), dump_name.size());
        buf_ptr += dump_name.size();
        memmove(buf_ptr, data.data(), data.size());
        desc = iter->second.first.create_desc(inputs.size(), buf.get(), buf_len);
    } else
        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);
    }

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