Skip to content

GitLab

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

change layout tests

parent 4e62ccf3
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
test/layout_nhwc.cpp
View file @ 07a0acdb
...@@ -76,8 +76,7 @@ TEST_CASE(conv_relu) ...@@ -76,8 +76,7 @@ TEST_CASE(conv_relu)
{{"padding", {1, 1}}, {"stride", {2, 2}}, {"dilation", {1, 1}}}), {{"padding", {1, 1}}, {"stride", {2, 2}}, {"dilation", {1, 1}}}),
x, x,
w); w);
auto relu = m2.add_instruction(migraphx::make_op("relu"), conv); auto conv_layout = m2.add_instruction(layout(), conv);
m2.add_instruction(layout(), relu);
} }
EXPECT(m1.sort() == m2.sort()); EXPECT(m1.sort() == m2.sort());
} }
...@@ -115,11 +114,12 @@ TEST_CASE(conv_add) ...@@ -115,11 +114,12 @@ TEST_CASE(conv_add)
{{"padding", {1, 1}}, {"stride", {2, 2}}, {"dilation", {1, 1}}}), {{"padding", {1, 1}}, {"stride", {2, 2}}, {"dilation", {1, 1}}}),
x, x,
w); w);
auto conv_layout = m2.add_instruction(layout(), conv);
auto b = m2.add_instruction( auto b = m2.add_instruction(
migraphx::make_op("broadcast", {{"axis", 1}, {"out_lens", conv->get_shape().lens()}}), migraphx::make_op("broadcast", {{"axis", 1}, {"out_lens", conv->get_shape().lens()}}),
y); y);
auto add = m2.add_instruction(migraphx::make_op("add"), conv, b); m2.add_instruction(migraphx::make_op("add"), conv_layout, b);
m2.add_instruction(layout(), add); // m2.add_instruction(layout(), add);
} }
EXPECT(m1.sort() == m2.sort()); 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