Dont sink when insert_slice

Signed-off-by: Ian Wood <ianwood2024@u.northwestern.edu>
iree-org · Dec 18, 2024 · f6c9d1d · f6c9d1d
1 parent 9c039e6
commit f6c9d1d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/compiler/src/iree/compiler/DispatchCreation/BubbleUpExpandShapes.cpp b/compiler/src/iree/compiler/DispatchCreation/BubbleUpExpandShapes.cpp
@@ -151,7 +151,10 @@ void BubbleUpExpandShapesPass::runOnOperation() {
         if (auto consumerLinalgOp = dyn_cast<linalg::LinalgOp>(consumer)) {
           return isa<linalg::GenericOp>(consumerLinalgOp) &&
                  llvm::all_of(consumerLinalgOp.getIteratorTypesArray(),
-                              linalg::isParallelIterator);
+                              linalg::isParallelIterator) &&
+                 !llvm::any_of(consumerLinalgOp->getUsers(), [](Operation *op) {
+                   return llvm::isa<tensor::InsertSliceOp>(op);
+                 });
         }
         // Fuse in all other cases.
         return true;