"vscode:/vscode.git/clone" did not exist on "ddd49ec9e7f829c3a4ca92de22206c708d8d94a7"
Commit 3544f085 authored by Paul's avatar Paul
Browse files

Finalize rocblas geem

parent c9f8c7a7
...@@ -36,6 +36,7 @@ struct run_op : action<run_op> ...@@ -36,6 +36,7 @@ struct run_op : action<run_op>
static void apply(const parser& p, const value& v) static void apply(const parser& p, const value& v)
{ {
context ctx; context ctx;
migraphx::context gctx = ctx;
auto inputs = p.parse_shapes(v.at("inputs")); auto inputs = p.parse_shapes(v.at("inputs"));
auto name = v.at("name").to<std::string>(); auto name = v.at("name").to<std::string>();
if(not contains(name, "::")) if(not contains(name, "::"))
...@@ -43,6 +44,10 @@ struct run_op : action<run_op> ...@@ -43,6 +44,10 @@ struct run_op : action<run_op>
auto op = make_op(name); auto op = make_op(name);
if(v.contains("fields")) if(v.contains("fields"))
op.from_value(v.at("fields")); op.from_value(v.at("fields"));
bool exhaustive = v.get("exhaustive", false);
ctx.set_exhaustive_tune_flag(exhaustive);
auto output = op.compute_shape(inputs);
op.finalize(gctx, output, inputs);
auto t = time_op(ctx, op, inputs, p.get(v, "iterations", 100)); auto t = time_op(ctx, op, inputs, p.get(v, "iterations", 100));
std::cout << op << ": " << t << "ms" << std::endl; std::cout << op << ": " << t << "ms" << std::endl;
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment