Method print_py

src/module.cpp:1363–1433

}

std::unordered_map<instruction_ref, std::string>
module::print_py(std::ostream& os,
                 const std::string& mname,
                 std::unordered_map<instruction_ref, std::string> names) const
{
    // cppcheck-suppress variableScope
    unsigned long seed = names.size();
    auto last          = std::prev(this->end());
    names              = this->print(
        [&](auto ins, auto ins_names) {
            std::vector<std::string> input_vars;
            std::transform(ins->inputs().begin(),
                           ins->inputs().end(),
                           std::back_inserter(input_vars),
                           [&](auto input) { return cpp_var_name(ins_names.at(input)); });
            if(ins != last)
                os << cpp_var_name(ins_names.at(ins)) << " = ";
            if(ins->name() == "@literal")
            {
                os << mname << ".add_literal(";
                if(ins->get_shape().elements() < 1024)
                {
                    os << "migraphx.create_argument(";
                    print_py_shape(os, ins->get_shape());
                    os << ", [" << ins->get_literal() << "])";
                }
                else
                {
                    const bool use_abs = false;
                    // Disable abs for now
                    // ins->get_literal().visit([&](auto v) {
                    //     use_abs = std::none_of(v.begin(), v.end(), [](auto x) { return x < 0; });
                    // });
                    if(use_abs)
                        os << "migraphx.abs_literal(";
                    os << "migraphx.generate_argument(";
                    print_py_shape(os, ins->get_shape());
                    os << ", " << seed << ")";
                    if(use_abs)
                        os << ")";
                    seed++;
                }
                os << ")" << std::endl;
            }
            else if(ins->name() == "@param")
            {
                std::string name = any_cast<builtin::param>(ins->get_operator()).parameter;
                os << mname << ".add_parameter(" << enclose_name(name) << ", ";
                print_py_shape(os, ins->get_shape());
                os << ")" << std::endl;
            }
            else if(ins->name() == "@return")
            {
                os << mname << ".add_return([" << join_strings(input_vars, ", ") << "])"
                   << std::endl;
            }
            else
            {