Method to_json

src/gopt/impl/global_layout_transform/subgraph_extractor.cpp:91–126

/* ================== GraphPartition =================*/
#if MGB_ENABLE_JSON
std::shared_ptr<json::Value> GraphPartition::to_json() const {
    auto replaced_outputs = std::get<1>(replace_graph_by_placeholder());

    ThinHashSet<VarNode*> all_var_node;
    ThinHashSet<OperatorNodeBase*> all_opr_node;
    auto comp_seq = json::Array::make();

    auto cb = [&](OperatorNodeBase* opr) {
        comp_seq->add(json::String::make(opr->id_str()));
        for (const auto& i : opr->input()) {
            if (all_var_node.count(i) == 0) {
                all_var_node.insert(i);
            }
        }
        all_opr_node.insert(opr);
        for (const auto& o : opr->output()) {
            all_var_node.insert(o);
        }
    };
    cg::DepOprIter iter{cb};
    for (const auto& o : replaced_outputs)
        iter.add(o->owner_opr());

    auto dump_node_coll = [](auto&& collection) {
        auto objptr = json::Object::make();
        auto&& obj = *objptr;
        for (auto&& i : collection)
            obj[i->id_str()] = i->to_json();
        return objptr;
    };

    return json::Object::make(
            {{"operator", dump_node_coll(all_opr_node)},
             {"var", dump_node_coll(all_var_node)},
             {"comp_seq", comp_seq}});
}
#endif

std::pair<VarNodeArray, VarNodeArray> GraphPartition::replace_graph_by_placeholder()