Method load

src/serialization/impl/serializer_oss.cpp:865–953

}

GraphLoader::LoadResult GraphLoaderOSS::load(const LoadConfig& config, bool rewind) {
    mgb_assert(m_file);
    m_cur_load_config = &config;
    if (rewind) {
        m_file->rewind();
    }
    uint32_t magic;
    m_file->read(&magic, sizeof(magic));
    mgb_throw_if(
            (magic != MGB_MAGIC) && (magic != MAGIC_V0), SerializationError,
            "wrong magic: wanted %#08x or %#08x, actual %#08x (not a invalid fbs "
            "model?)",
            MGB_MAGIC, MAGIC_V0, magic);
    if (magic == MGB_MAGIC) {
        // read FeatureBits
        magic_compare = true;
        m_file->read(&m_feature_bits, 8);
    } else {
        magic_compare = false;
    }
    m_file->skip(4);

    uint64_t offset_to_fbs;
    m_file->read(&offset_to_fbs, sizeof(offset_to_fbs));
    auto tensor_begin = m_file->tell();
    // Skip tensor data
    m_file->skip(offset_to_fbs);

    // Read fbs::Graph
    uint32_t size;
    m_file->read(&size, sizeof(size));
    m_file->skip(-sizeof(size));
    m_graph_buf = m_file->read_shared(size + sizeof(size));

    // Rewind back to tensor data
    m_file->rewind();
    m_file->skip(tensor_begin);

    {
        flatbuffers::Verifier verifier(
                static_cast<const uint8_t*>(m_graph_buf.data()), m_graph_buf.size());
        mgb_throw_if(
                !fbs::VerifySizePrefixedGraphBuffer(verifier), SerializationError,
                "model verification failed (invalid or corrupted model?)");
    }

    m_graph = fbs::GetSizePrefixedGraph(m_graph_buf.data());
    m_mgb_version = m_graph->mgb_version();
    if (m_graph->mgb_version() > MGB_VERSION) {
        mgb_log_warn(
                "loading model from future runtime: version=%u "
                "model_version=%u",
                MGB_VERSION, m_graph->mgb_version());
    }
    if (!m_graph_hash) {
        m_graph_hash = m_graph->hash();
        mgb_assert(
                m_graph_hash,