Commit 3de56715 authored by wsttiger's avatar wsttiger
Browse files

Formatting

parent 447eec93
......@@ -8,10 +8,12 @@ namespace migraph {
namespace gpu {
namespace device {
argument concat(const migraph::shape& output_shape, std::vector<migraph::argument> args, std::vector<std::size_t> offsets)
argument concat(const migraph::shape& output_shape,
std::vector<migraph::argument> args,
std::vector<std::size_t> offsets)
{
//migraph::argument& result = args.back();
for(std::size_t l = 0; l < args.size()-1; l++)
// migraph::argument& result = args.back();
for(std::size_t l = 0; l < args.size() - 1; l++)
{
auto argl = args[l];
std::size_t nelements = argl.get_shape().elements();
......@@ -21,17 +23,15 @@ argument concat(const migraph::shape& output_shape, std::vector<migraph::argumen
const auto* inptr = input.data();
hip_tensor_descriptor<ndim> desc_input(input.get_shape());
hip_tensor_descriptor<ndim> desc_output(output.get_shape());
gs_launch(nelements)([=](auto i) {
outptr[desc_output.linear(desc_input.multi(i))] = inptr[i];
});
gs_launch(nelements)(
[=](auto i) { outptr[desc_output.linear(desc_input.multi(i))] = inptr[i]; });
});
});
}
//return result;
// return result;
return args.back();
}
} // namespace device
} // namespace gpu
} // namespace migraph
......@@ -5,7 +5,8 @@ namespace migraph {
namespace gpu {
namespace device {
argument concat(const shape& output_shape, std::vector<argument> args, std::vector<std::size_t> offsets);
argument
concat(const shape& output_shape, std::vector<argument> args, std::vector<std::size_t> offsets);
} // namespace device
} // namespace gpu
......
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