Commit 2228899a authored by Shucai Xiao's avatar Shucai Xiao
Browse files

change the gpu slice test according to comments.

parent 5f82df2f
...@@ -221,15 +221,11 @@ struct test_slice ...@@ -221,15 +221,11 @@ struct test_slice
migraph::program create_program() const migraph::program create_program() const
{ {
migraph::program p; migraph::program p;
migraph::shape s{migraph::shape::int32_type, {2, 2, 3}}; migraph::shape s{migraph::shape::int32_type, {2, 2, 4}};
std::vector<int> data(2 * 2 * 3); auto x = p.add_parameter("x", s);
std::iota(data.begin(), data.end(), 0); auto y = p.add_parameter("y", {migraph::shape::int32_type, {2, 2, 2}});
auto l = p.add_literal(migraph::literal{s, data}); auto slice0 = p.add_instruction(migraph::op::slice{{2}, {0}, {2}}, x);
auto slice0 = p.add_instruction(migraph::op::slice{{2}, {0}, {1}}, l); p.add_instruction(migraph::op::add{}, y, slice0);
auto slice1 = p.add_instruction(migraph::op::slice{{2}, {1}, {3}}, l);
auto add0 = p.add_instruction(migraph::op::add{}, slice0, slice0);
auto add1 = p.add_instruction(migraph::op::add{}, slice1, slice1);
p.add_instruction(migraph::op::concat{2}, add0, add1);
return p; return p;
} }
......
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