Commit 07a0acdb authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

change layout tests

parent 4e62ccf3
......@@ -76,8 +76,7 @@ TEST_CASE(conv_relu)
{{"padding", {1, 1}}, {"stride", {2, 2}}, {"dilation", {1, 1}}}),
x,
w);
auto relu = m2.add_instruction(migraphx::make_op("relu"), conv);
m2.add_instruction(layout(), relu);
auto conv_layout = m2.add_instruction(layout(), conv);
}
EXPECT(m1.sort() == m2.sort());
}
......@@ -115,11 +114,12 @@ TEST_CASE(conv_add)
{{"padding", {1, 1}}, {"stride", {2, 2}}, {"dilation", {1, 1}}}),
x,
w);
auto conv_layout = m2.add_instruction(layout(), conv);
auto b = m2.add_instruction(
migraphx::make_op("broadcast", {{"axis", 1}, {"out_lens", conv->get_shape().lens()}}),
y);
auto add = m2.add_instruction(migraphx::make_op("add"), conv, b);
m2.add_instruction(layout(), add);
m2.add_instruction(migraphx::make_op("add"), conv_layout, b);
// m2.add_instruction(layout(), add);
}
EXPECT(m1.sort() == m2.sort());
}
......
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