Skip to content

GitLab

Unverified Commit 193f105d authored by Charlie Lin's avatar Charlie Lin Committed by GitHub
Browse files

Fix shape typo in API test (#1787)

parent 0dae73fa
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
test/api/test_gpu.cpp
View file @ 193f105d
...@@ -151,7 +151,7 @@ TEST_CASE(dynamic_batch_load_and_run_offload) ...@@ -151,7 +151,7 @@ TEST_CASE(dynamic_batch_load_and_run_offload)
c_options.set_offload_copy(); c_options.set_offload_copy();
p.compile(migraphx::target("gpu"), c_options); p.compile(migraphx::target("gpu"), c_options);
auto out_shapes = p.get_output_shapes(); auto out_shapes = p.get_output_shapes();
CHECK(out_shapes.size() == 1); EXPECT(out_shapes.size() == 1);
EXPECT(out_shapes[0].dynamic()); EXPECT(out_shapes[0].dynamic());
// batch size = 2 // batch size = 2
...@@ -165,9 +165,9 @@ TEST_CASE(dynamic_batch_load_and_run_offload) ...@@ -165,9 +165,9 @@ TEST_CASE(dynamic_batch_load_and_run_offload)
migraphx::argument(migraphx::shape(migraphx_shape_float_type, {2, 3, 3, 3}), c.data())); migraphx::argument(migraphx::shape(migraphx_shape_float_type, {2, 3, 3, 3}), c.data()));
auto outputs = p.eval(pp); auto outputs = p.eval(pp);
CHECK(shapes_before.size() == outputs.size()); EXPECT(shapes_before.size() == outputs.size());
CHECK(bool{outputs.front().get_shape() == EXPECT(bool{outputs.front().get_shape() ==
migraphx::shape(migraphx_shape_float_type, {2, 1, 3, 3})}); migraphx::shape(migraphx_shape_float_type, {2, 2, 2, 2})});
} }
TEST_CASE(load_and_run_async) TEST_CASE(load_and_run_async)
......
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