Commit cb4e7a5a authored by Brian Pickrell's avatar Brian Pickrell
Browse files

comments

parent 5ec7b014
...@@ -23,9 +23,9 @@ ...@@ -23,9 +23,9 @@
*/ */
/** /**
* Parent struct for prefix scans. A prefix scan is a mathematical entity useful * Parent struct for prefix scan ops. A prefix scan is a mathematical entity useful
* in parallelizing various computations. Given a list of numbers, a prefix scan * in parallelizing various computations. Given a list of numbers, a prefix scan
* is an equal size list of running totals of the values. Other operations * op returns an equal size list of running totals of the values. Other operations
* besides addition can be supported by child ops. * besides addition can be supported by child ops.
*/ */
#ifndef MIGRAPHX_GUARD_OPERATORS_SCAN_OP_HPP #ifndef MIGRAPHX_GUARD_OPERATORS_SCAN_OP_HPP
......
...@@ -5546,12 +5546,13 @@ TEST_CASE(prefix_scan_sum_dyn_1d) ...@@ -5546,12 +5546,13 @@ TEST_CASE(prefix_scan_sum_dyn_1d)
std::vector<migraphx::shape::dynamic_dimension> dd{{5, 8}}; std::vector<migraphx::shape::dynamic_dimension> dd{{5, 8}};
migraphx::shape s{migraphx::shape::float_type, dd}; migraphx::shape s{migraphx::shape::float_type, dd};
auto input = mm->add_parameter("X", s); auto input = mm->add_parameter("X", s);
mm->add_instruction(migraphx::make_op("prefix_scan_sum", {{"axis", 0}, {"exclusive", false}}), input); mm->add_instruction(migraphx::make_op("prefix_scan_sum", {{"axis", 0}, {"exclusive", false}}),
input);
p.compile(migraphx::make_target("ref")); p.compile(migraphx::make_target("ref"));
std::vector<float> a = {1, 2, 3, 4, 5, 6}; std::vector<float> a = {1, 2, 3, 4, 5, 6};
migraphx::shape input_fixed_shape0{migraphx::shape::float_type, {6}}; migraphx::shape input_fixed_shape0{migraphx::shape::float_type, {6}};
migraphx::parameter_map params0; migraphx::parameter_map params0;
params0["X"] = migraphx::argument(input_fixed_shape0, a.data()); params0["X"] = migraphx::argument(input_fixed_shape0, a.data());
auto result = p.eval(params0).back(); auto result = p.eval(params0).back();
......
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