Function find_pointwise_modules

src/fuse_pointwise.cpp:274–306

}

static bool find_pointwise_modules(module_pass_manager& mpm, bool multi_out)
{
    bool changed = false;
    auto last    = std::prev(mpm.get_module().end());
    for(auto ins : iterator_for(mpm.get_module()))
    {
        if(ins != last and is_dead(ins))
            continue;
        auto pw_outs = find_output_pointwise(mpm.get_module(), ins, multi_out);

        if(pw_outs.size() > 1)
        {
            (void)std::accumulate(
                pw_outs.begin() + 1, pw_outs.end(), pw_outs.front(), [&](auto input, auto output) {
                    return merge_instruction(mpm, input, output);
                });
            changed = true;
        }
        else if(ins->name() == "pointwise")
        {
            auto it = find_input_pointwise(mpm.get_module(), ins, multi_out);
            if(it == ins->inputs().end())
                continue;
            auto input = *it;
            if(is_dead(input))
                continue;
            merge_instruction(mpm, input, ins);

            changed = true;
        }
    }
    return changed;
}

namespace {
struct pointwise_reshape : rewrite_reshapes_base