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

Formatting

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