Method apply

src/fuse_pointwise_reduce.cpp:46–59

}

void fuse_pointwise_reduce::apply(module_pass_manager& mpm) const
{
    mpm.run_pass(fuse_pointwise{.enable_rewrite_reshapes = false});
    mpm.run_pass(optimize_module{});
    mpm.run_pass(fuse_reduce{.enable_rewrite_reshapes = false});
    mpm.run_pass(fuse_pointwise{.enable_rewrite_reshapes = true});
    mpm.run_pass(fuse_reduce{.enable_rewrite_reshapes = true});
    mpm.run_pass(split_reduce{.split_size = get_split_size(split_size)});
    mpm.run_pass(fuse_pointwise{.enable_rewrite_broadcasts = true});
    if(not enabled(MIGRAPHX_DISABLE_MULTI_OUTPUT_FUSION{}))
    {
        mpm.run_pass(fuse_pointwise{.enable_multi_output = true});
    }
}

} // namespace MIGRAPHX_INLINE_NS
} // namespace migraphx