[Pass] Fix non constant loop bound in .outline()

cornell-zhang · Aug 8, 2022 · 6a856ef · 6a856ef
1 parent ec69284
commit 6a856ef
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/Transforms/LoopTransformations.cpp b/lib/Transforms/LoopTransformations.cpp
@@ -2534,7 +2534,9 @@ LogicalResult runOutline(ModuleOp &mod, func::FuncOp &f, OutlineOp &outlineOp) {
             assert(idx != -1 && "Not found target IV");
             loops.push_back({targetLoop, idx});
           } else {
-            assert(false && "Not supported");
+            assert(srcLoop.getUpperBound().getMap() ==
+                       targetLoop.getUpperBound().getMap() &&
+                   "map mismatch");
           }
         }
       }