Commit 3c733048 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

refine implementation

parent 255d6868
...@@ -19,6 +19,10 @@ void auto_contiguous::apply(module& p) const ...@@ -19,6 +19,10 @@ void auto_contiguous::apply(module& p) const
auto args = ins->inputs(); auto args = ins->inputs();
auto new_args = args; auto new_args = args;
std::transform(args.begin(), args.end(), new_args.begin(), [&](auto in) { std::transform(args.begin(), args.end(), new_args.begin(), [&](auto in) {
if (in->name() == "contiguous")
{
return in;
}
return p.insert_instruction(ins, make_op("contiguous"), in); return p.insert_instruction(ins, make_op("contiguous"), in);
}); });
...@@ -32,6 +36,7 @@ void auto_contiguous::apply(module& p) const ...@@ -32,6 +36,7 @@ void auto_contiguous::apply(module& p) const
auto last = std::prev(p.end()); auto last = std::prev(p.end());
for(auto ins : iterator_for(p)) for(auto ins : iterator_for(p))
{ {
// for last instruction that is NOT a return
if(ins->outputs().empty() and ins != last) if(ins->outputs().empty() and ins != last)
continue; continue;
shape s = ins->get_shape(); shape s = ins->get_shape();
......
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