Skip to content

GitLab

Commit 0f91b993 authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

change layout tests

parent 90071be5
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
test/layout_nhwc.cpp
View file @ 0f91b993
...@@ -76,7 +76,8 @@ TEST_CASE(conv_relu) ...@@ -76,7 +76,8 @@ 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);
m2.add_instruction(layout(), conv); auto conv_layout = m2.add_instruction(layout(), conv);
m2.add_instruction(migraphx::make_op("relu"), conv_layout);
} }
EXPECT(m1.sort() == m2.sort()); EXPECT(m1.sort() == m2.sort());
} }
...@@ -115,7 +116,7 @@ TEST_CASE(conv_add) ...@@ -115,7 +116,7 @@ TEST_CASE(conv_add)
x, x,
w); w);
auto conv_layout = m2.add_instruction(layout(), conv); 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);
m2.add_instruction(migraphx::make_op("add"), conv_layout, b); m2.add_instruction(migraphx::make_op("add"), conv_layout, b);
......
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