Function time_program

src/targets/gpu/time_op.cpp:106–132

}

double time_program(const context& ictx,
                    program p,
                    const std::unordered_map<std::string, double>& fill_map,
                    int bundle,
                    int nruns)
{
    std::vector<migraphx::context> ctx_vec = {ictx};
    auto& gctx                             = any_cast<migraphx::gpu::context>(ctx_vec.front());
    auto* mm                               = p.get_main_module();
    mm->finalize(ctx_vec);
    auto in_shapes = p.get_parameter_shapes();
    std::unordered_map<std::string, migraphx::argument> param_map;
    unsigned long seed = 0;
    for(const auto& [name, shape] : in_shapes)
    {
        std::string id = "";
        if(shape.type() != migraphx::shape::tuple_type)
            id = shape.type_string() + migraphx::shape::to_sizes_string({shape.as_standard()});

        if(contains(fill_map, id))
            param_map[name] = to_gpu(fill_argument(shape, fill_map.at(id)));
        else
            param_map[name] = to_gpu(generate_argument(shape, seed++, random_mode::random));
    }
    auto run = [&] { p.eval_with_context(ctx_vec, param_map); };
    return time_loop(gctx, bundle, nruns, run);
}

} // namespace gpu
} // namespace MIGRAPHX_INLINE_NS