"vscode:/vscode.git/clone" did not exist on "e4a0bd2ceb6c80ec90194ce519bbd9589926034b"
Commit 7995cbb8 authored by Paul's avatar Paul
Browse files

Rename to pack

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