Function append_pointwise_module

src/fuse_pointwise.cpp:141–163

}

static module::with_inputs append_pointwise_module(instruction_ref ins, instruction_ref output)
{
    std::unordered_set<instruction_ref> original_inputs{ins->inputs().begin(), ins->inputs().end()};
    original_inputs.insert(output->inputs().begin(), output->inputs().end());
    module pm     = *ins->module_inputs().at(0);
    module_ref xm = output->module_inputs().at(0);
    const bool dependent = contains(output->inputs(), ins);
    assert(not dependent or pm.get_returns().size() == 1);

    std::unordered_map<instruction_ref, instruction_ref> map_ins =
        pm.get_ins_param_map(ins->inputs());
    if(dependent)
        map_ins[ins] = pm.get_returns().front();
    auto returns = pm.fuse(*xm, output->inputs(), &map_ins, nullptr, &to_scalar);
    if(not is_used_once(ins, &output) or not dependent)
    {
        auto ireturns = pm.get_returns();
        returns.insert(returns.end(), ireturns.begin(), ireturns.end());
    }
    pm.replace_return(returns);
    auto inputs = find_inputs(map_ins, original_inputs, &pm);
    return {std::move(pm), inputs};
}

static void replace_with_tuple(module& m, instruction_ref ins, instruction_ref rep, bool first)