Commit 56a6b232 authored by Paul's avatar Paul
Browse files

Use func.return

parent 516779cb
......@@ -417,7 +417,7 @@ struct mlir_program
static std::string get_name(instruction_ref ins)
{
if(ins->name() == "@return")
return "std.return";
return "func.return";
return "migraphx." + ins->name();
}
......
......@@ -109,7 +109,8 @@ module {
migraphx::module m;
auto x = m.add_parameter("x", {migraphx::shape::float_type, {1, 8, 4, 4}});
auto w = m.add_parameter("w", {migraphx::shape::float_type, {2, 8, 3, 3}});
auto conv = m.add_instruction(migraphx::make_op("convolution"), x, w);
auto conv = m.add_instruction(migraphx::make_op("convolution", {{"padding", {0, 0, 0, 0}}}), x, w);
// auto conv = m.add_instruction(migraphx::make_op("convolution"), x, w);
m.add_return({conv});
auto s = migraphx::gpu::dump_mlir(m);
// Skip test if MLIR is not enabled
......
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