Function create_program_from_mlir

test/gpu/mlir.cpp:90–109

}

static migraphx::program create_program_from_mlir(const migraphx::module& mmlir)
{
    migraphx::program p;
    auto* mm   = p.get_main_module();
    auto names = mmlir.get_parameter_names();
    std::vector<migraphx::instruction_ref> inputs;
    std::transform(names.begin(), names.end(), std::back_inserter(inputs), [&](const auto& name) {
        return mm->add_parameter(name, mmlir.get_parameter_shape(name));
    });
    std::sort(inputs.begin(), inputs.end(), migraphx::by(std::less<>{}, [](auto ins) {
                  return to_string(ins->get_operator());
              }));
    inputs.push_back(mm->add_parameter("output", mmlir.get_output_shapes().front()));

    migraphx::gpu::context ctx;
    migraphx::gpu::mlir_code_object mco =
        compile_mlir(ctx, create_mlir_submodule(mmlir), to_shapes(inputs), {});
    migraphx::gpu::insert_mlir(*mm, mm->end(), mco.cop, inputs);
    return p;
}

static migraphx::parameter_map generate_params(const migraphx::program& p)
{