Commit 2863aa3e authored by Paul's avatar Paul
Browse files

Formatting

parent 632f819e
...@@ -12,17 +12,17 @@ ...@@ -12,17 +12,17 @@
namespace migraphx { namespace migraphx {
inline namespace MIGRAPHX_INLINE_NS { inline namespace MIGRAPHX_INLINE_NS {
template<class Predicate> template <class Predicate>
std::vector<instruction_ref> find_lasts(const module& m, Predicate pred) std::vector<instruction_ref> find_lasts(const module& m, Predicate pred)
{ {
std::vector<instruction_ref> result; std::vector<instruction_ref> result;
fix([&](auto self, auto ins) { fix([&](auto self, auto ins) {
if (pred(ins)) if(pred(ins))
{ {
result.push_back(ins); result.push_back(ins);
return; return;
} }
for(auto input:ins->inputs()) for(auto input : ins->inputs())
self(input); self(input);
})(std::prev(m.end())); })(std::prev(m.end()));
return result; return result;
...@@ -30,18 +30,17 @@ std::vector<instruction_ref> find_lasts(const module& m, Predicate pred) ...@@ -30,18 +30,17 @@ std::vector<instruction_ref> find_lasts(const module& m, Predicate pred)
void preserve_output_layout(module& m) void preserve_output_layout(module& m)
{ {
std::vector<instruction_ref> outputs = find_lasts(m, [](auto ins) { std::vector<instruction_ref> outputs =
return ins->get_shape().lens().size() == 4; find_lasts(m, [](auto ins) { return ins->get_shape().lens().size() == 4; });
}); for(auto output : outputs)
for(auto output:outputs)
{ {
auto permutation = find_permutation(output->get_shape()); auto permutation = find_permutation(output->get_shape());
auto layout = m.insert_instruction(std::next(output), make_op("layout", {{"permutation", permutation}}), output); auto layout = m.insert_instruction(
std::next(output), make_op("layout", {{"permutation", permutation}}), output);
m.replace_instruction(output, layout); m.replace_instruction(output, layout);
} }
} }
void transform_convolutions(module& m) void transform_convolutions(module& m)
{ {
for(auto ins : iterator_for(m)) for(auto ins : iterator_for(m))
......
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