Method get_split_point_set

src/core/impl/graph/seq_sublinear_memory.cpp:125–170

}

SeqModifierForSublinearMemory::SplitPointSet SeqModifierForSublinearMemory::
        ModifyActionPlanner::get_split_point_set(size_t block_size_thresh) {
    auto split_point_set = make_split_point_set();
    size_t cur_block_usage = 0;

    ThinHashSet<Var*> cur_block_alive_vars;

    auto add_alive = [&](Var* var) {
        auto&& ins = cur_block_alive_vars.insert(var);
        mgb_assert(ins.second);
        cur_block_usage += var->size;
    };

    auto remove_alive = [&](Var* var) {
        if (cur_block_alive_vars.erase(var)) {
            auto size = var->size;
            mgb_assert(size <= cur_block_usage);
            cur_block_usage -= size;
        }
    };

    auto flush_block_member = [&](size_t p) {
        split_point_set->push_back(p);
        cur_block_usage = 0;
        cur_block_alive_vars.clear();
    };

    for (size_t i = 0; i < seq().size(); ++i) {
        auto opr = seq()[i].get();

        for (auto i : opr->output)
            add_alive(i);

        for (auto i : opr->input) {
            if (opr == i->last_access_opr())
                remove_alive(i);
        }

        if (i + 1 < seq().size() && (cur_block_usage < block_size_thresh ||
                                     (nr_endpoint_oprs() && !opr->is_endpoint)))
            continue;

        flush_block_member(i);
    }
    return split_point_set;
}

size_t SeqModifierForSublinearMemory::ModifyActionPlanner::
        calc_bottleneck_from_discard_plan() {