Commit 7995cbb8 authored by Paul's avatar Paul
Browse files

Rename to pack

parent 6a2d72d7
...@@ -35,7 +35,7 @@ auto fix(F f) ...@@ -35,7 +35,7 @@ auto fix(F f)
} }
template <class... Ts> template <class... Ts>
auto make_sequence(Ts... xs) auto pack(Ts... xs)
{ {
return [=](auto f) { return f(xs...); }; return [=](auto f) { return f(xs...); };
} }
......
...@@ -16,7 +16,7 @@ auto nary_nonstandard_impl(F f, argument result, Arguments... args) ...@@ -16,7 +16,7 @@ auto nary_nonstandard_impl(F f, argument result, Arguments... args)
const auto& output_shape = result.get_shape(); const auto& output_shape = result.get_shape();
visit_all(result, args...)([&](auto output, auto... inputs) { visit_all(result, args...)([&](auto output, auto... inputs) {
visit_tensor_size(output_shape.lens().size(), [&](auto ndim) { visit_tensor_size(output_shape.lens().size(), [&](auto ndim) {
auto data = make_sequence( auto data = pack(
std::make_pair(hip_tensor_descriptor<ndim>{inputs.get_shape().lens(), std::make_pair(hip_tensor_descriptor<ndim>{inputs.get_shape().lens(),
inputs.get_shape().strides()}, inputs.get_shape().strides()},
inputs.data())...); inputs.data())...);
...@@ -45,7 +45,7 @@ auto nary_standard(argument result, Arguments... args) ...@@ -45,7 +45,7 @@ auto nary_standard(argument result, Arguments... args)
// assert(x.get_shape().elements() == y.get_shape().elements()); // assert(x.get_shape().elements() == y.get_shape().elements());
const auto& output_shape = result.get_shape(); const auto& output_shape = result.get_shape();
visit_all(result, args...)([&](auto output, auto... inputs) { visit_all(result, args...)([&](auto output, auto... inputs) {
auto data = make_sequence(inputs.data()...); auto data = pack(inputs.data()...);
auto* outp = output.data(); auto* outp = output.data();
gs_launch(output_shape.elements())( gs_launch(output_shape.elements())(
[=](auto i) { data([&](auto... xps) { outp[i] = f(xps[i]...); }); }); [=](auto i) { data([&](auto... xps) { outp[i] = f(xps[i]...); }); });
......
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