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

Use func.return

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