Commit 3855b681 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

refine the adjust_allocation pass.

parent 6c1a08f4
...@@ -47,7 +47,7 @@ TEST_CASE(tanh_shape) ...@@ -47,7 +47,7 @@ TEST_CASE(tanh_shape)
EXPECT(p1 == p2); EXPECT(p1 == p2);
p1.compile(lowering_target{}); p1.compile(lowering_target{});
p2.compile(lowering_target{}); p2.compile(lowering_target());
EXPECT(p1 == p2); EXPECT(p1 == p2);
...@@ -55,13 +55,13 @@ TEST_CASE(tanh_shape) ...@@ -55,13 +55,13 @@ TEST_CASE(tanh_shape)
{ {
if(ins->name() == "hip::allocate") if(ins->name() == "hip::allocate")
{ {
migraphx::shape wrong_s{migraphx::shape::float_type, {2, 2}}; migraphx::shape wrong_s{migraphx::shape::float_type, {3, 2}, {1, 3}};
ins->replace(wrong_s); ins->replace(wrong_s);
} }
} }
EXPECT(p1 != p2); EXPECT(p1 != p2);
migraphx::run_passes(p1, migraphx::run_passes(p2,
{migraphx::gpu::adjust_allocation{}, migraphx::dead_code_elimination{}}); {migraphx::gpu::adjust_allocation{}, migraphx::dead_code_elimination{}});
EXPECT(p1 == p2); EXPECT(p1 == p2);
} }
......
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