Commit 45bdfaa0 authored by Paul's avatar Paul
Browse files

Formatting

parent 6812d12c
...@@ -3,23 +3,25 @@ ...@@ -3,23 +3,25 @@
namespace migraph { namespace migraph {
template <class T> template <class T>
struct iterator_for_range { struct iterator_for_range
{
T* base; T* base;
using base_iterator = decltype(base->begin()); using base_iterator = decltype(base->begin());
struct iterator { struct iterator
{
base_iterator i; base_iterator i;
base_iterator operator * () { return i; } base_iterator operator*() { return i; }
base_iterator operator ++ () { return ++i; } base_iterator operator++() { return ++i; }
bool operator != (const iterator& rhs) { return i != rhs.i; } bool operator!=(const iterator& rhs) { return i != rhs.i; }
}; };
iterator begin() { return {base->begin()}; } iterator begin() { return {base->begin()}; }
iterator end() { return {base->end()}; } iterator end() { return {base->end()}; }
}; };
template <class T> template <class T>
iterator_for_range<T> iterator_for(T& x) iterator_for_range<T> iterator_for(T& x)
{ {
return {&x}; return {&x};
} }
......
...@@ -76,20 +76,17 @@ struct id_target ...@@ -76,20 +76,17 @@ struct id_target
struct reverse_pass struct reverse_pass
{ {
std::string name() const std::string name() const { return "reverse_pass"; }
{
return "reverse_pass";
}
void apply(migraph::program& p) const void apply(migraph::program& p) const
{ {
for(auto ins:migraph::iterator_for(p)) for(auto ins : migraph::iterator_for(p))
{ {
if(ins->op.name() == "sum") if(ins->op.name() == "sum")
{ {
p.replace_instruction(ins, minus_op{}, ins->arguments); p.replace_instruction(ins, minus_op{}, ins->arguments);
} }
else if(ins->op.name() == "minus") else if(ins->op.name() == "minus")
{ {
p.replace_instruction(ins, sum_op{}, ins->arguments); p.replace_instruction(ins, sum_op{}, ins->arguments);
} }
...@@ -100,19 +97,16 @@ struct reverse_pass ...@@ -100,19 +97,16 @@ struct reverse_pass
struct reverse_target struct reverse_target
{ {
std::string name() const { return "reverse"; } std::string name() const { return "reverse"; }
std::vector<migraph::pass> get_passes(migraph::context&) const std::vector<migraph::pass> get_passes(migraph::context&) const { return {reverse_pass{}}; }
{
return { reverse_pass{} };
}
migraph::context get_context() const { return {}; } migraph::context get_context() const { return {}; }
}; };
struct double_reverse_target struct double_reverse_target
{ {
std::string name() const { return "double_reverse"; } std::string name() const { return "double_reverse"; }
std::vector<migraph::pass> get_passes(migraph::context&) const std::vector<migraph::pass> get_passes(migraph::context&) const
{ {
return { reverse_pass{}, reverse_pass{} }; return {reverse_pass{}, reverse_pass{}};
} }
migraph::context get_context() const { return {}; } migraph::context get_context() const { return {}; }
}; };
......
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