Function main

src/driver/main.cpp:1005–1066

NOLINT

using namespace migraphx::driver; // NOLINT
int main(int argc, const char* argv[], const char* envp[])
{
    std::vector<std::string> args(argv + 1, argv + argc);
    // no argument, print the help infomration by default
    if(args.empty())
    {
        args.push_back("-h");
    }

    auto&& m = get_commands();
    auto cmd = args.front();

    if(cmd == "--ort-sha")
    {
        std::cout << MIGRAPHX_ORT_SHA1 << std::endl;
        return 0;
    }
    if(cmd == "-v" or cmd == "--version")
    {
        std::cout << get_version() << std::endl;
        return 0;
    }

    if(m.count(cmd) > 0)
    {
        std::string driver_invocation =
            std::string(argv[0]) + " " + migraphx::to_string_range(args, " ");
        std::cout << "Running [ " << get_version() << " ]: " << driver_invocation << std::endl;

        // Print start timestamp
        auto start_time = std::chrono::system_clock::now();
        std::cout << "[" << get_formatted_timestamp(start_time) << "]" << std::endl;

        m.at(cmd)(argv[0],
                  {args.begin() + 1, args.end()}); // run driver command found in commands map

        // Dump all the MIGraphX (consumed) Environment Variables:
        const auto mgx_env_map = migraphx::get_all_envs();
        for(auto&& [k, v] : mgx_env_map)
            std::cout << k << "=" << v << "\\ \n"; // backslash(s) to facilitate cut-n-paste

        auto unused_envs = get_unrecognized_migraphx_envs(envp, mgx_env_map);
        for(auto&& e : unused_envs)
            std::cout << "Unused environment variable: " << e << "\n";

        // Print end timestamp
        auto end_time = std::chrono::system_clock::now();
        std::cout << "[" << get_formatted_timestamp(end_time) << "]" << std::endl;

        // Print total duration
        auto duration =
            std::chrono::duration_cast<std::chrono::duration<double>>(end_time - start_time);
        std::cout << "[ " << get_version() << " ] Complete(" << duration.count()
                  << "s): " << driver_invocation << std::endl;
    }
    else
    {
        run_command<main_command>(argv[0], args);