"benchmark/vscode:/vscode.git/clone" did not exist on "fb8b3afa0a222d4a93ac2497d104769d4156dae9"
Commit 36f76f90 authored by Brian Pickrell's avatar Brian Pickrell
Browse files

formatting

parent 57d000b2
...@@ -108,7 +108,7 @@ struct reduce_op : op_name<Derived> ...@@ -108,7 +108,7 @@ struct reduce_op : op_name<Derived>
/** /**
* @brief returns a shape in which the axis or axes named * @brief returns a shape in which the axis or axes named
* for reduction by this op are set to size 1. * for reduction by this op are set, to size 1.
* *
* @param inputs list of input shapes * @param inputs list of input shapes
* @return shape * @return shape
......
...@@ -141,7 +141,7 @@ struct shape ...@@ -141,7 +141,7 @@ struct shape
* Same as the number of indices required to get a data value. * Same as the number of indices required to get a data value.
*/ */
std::size_t ndim() const; std::size_t ndim() const;
/*! /*!
* Return the number of elements in the tensor. * Return the number of elements in the tensor.
*/ */
......
...@@ -4241,13 +4241,14 @@ TEST_CASE(reducel1_test) ...@@ -4241,13 +4241,14 @@ TEST_CASE(reducel1_test)
EXPECT(p == prog); EXPECT(p == prog);
} }
TEST_CASE(reducel1_dyn_test) TEST_CASE(reducel1_dyn_test)
{ {
migraphx::program p; migraphx::program p;
auto* mm = p.get_main_module(); auto* mm = p.get_main_module();
// a shape with 4 dynamic dimensions // a shape with 4 dynamic dimensions
auto l0 = mm->add_parameter("x", migraphx::shape{migraphx::shape::float_type, {{3,3,0}, {3, 5, 0}, {4, 6, 5}, {5, 7, 6}}}); auto l0 = mm->add_parameter(
"x",
migraphx::shape{migraphx::shape::float_type, {{3, 3, 0}, {3, 5, 0}, {4, 6, 5}, {5, 7, 6}}});
auto abs_l0 = mm->add_instruction(migraphx::make_op("abs"), l0); auto abs_l0 = mm->add_instruction(migraphx::make_op("abs"), l0);
auto sum_l0 = mm->add_instruction(migraphx::make_op("reduce_sum", {{"axes", {-2}}}), abs_l0); auto sum_l0 = mm->add_instruction(migraphx::make_op("reduce_sum", {{"axes", {-2}}}), abs_l0);
mm->add_instruction(migraphx::make_op("squeeze", {{"axes", {-2}}}), sum_l0); mm->add_instruction(migraphx::make_op("squeeze", {{"axes", {-2}}}), sum_l0);
......
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