Skip last operator in simplify reshapes

0836b544 · Paul · d2968df8 · 0836b544
Commit 0836b544 authored May 01, 2019 by Paul
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

src/simplify_reshapes.cpp src/simplify_reshapes.cpp +3 -0

No files found.
--- a/src/simplify_reshapes.cpp
+++ b/src/simplify_reshapes.cpp
@@ -47,6 +47,9 @@ void simplify_reshapes::apply(program& p) const
    auto end = std::prev(p.end());
    for(auto ins : iterator_for(p))
    {
+        if (ins == end and ins->name() == "contiguous")
+            continue;
+        // Skip possible dead instructions
        if(ins->outputs().empty() and ins != end)
            continue;
        if(is_reshaper(ins))