Commit ee382ad9 authored by Paul's avatar Paul
Browse files

Format

parent 2a0ff223
...@@ -398,14 +398,14 @@ struct mlir_program ...@@ -398,14 +398,14 @@ 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 "std.return";
return "migraphx." + ins->name(); return "migraphx." + ins->name();
} }
static shape get_shape(instruction_ref ins) static shape get_shape(instruction_ref ins)
{ {
if (ins->name() == "@return") if(ins->name() == "@return")
{ {
assert(ins->inputs().size() == 1); assert(ins->inputs().size() == 1);
return ins->inputs().front()->get_shape(); return ins->inputs().front()->get_shape();
......
...@@ -40,8 +40,8 @@ si64]} : (tensor<1x8x4x4xf32>, tensor<2x8x3x3xf32>) -> tensor<1x2x2x2xf32> ...@@ -40,8 +40,8 @@ si64]} : (tensor<1x8x4x4xf32>, tensor<2x8x3x3xf32>) -> tensor<1x2x2x2xf32>
} }
)__migraphx__"; )__migraphx__";
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"), x, w);
m.add_return({conv}); m.add_return({conv});
auto s = migraphx::gpu::dump_mlir(m); auto s = migraphx::gpu::dump_mlir(m);
......
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