Commit ee382ad9 authored by Paul's avatar Paul
Browse files

Format

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