Dont sink when insert_slice

iree-org · Nov 13, 2024 · b9ba438 · b9ba438
1 parent 7bd88a7
commit b9ba438
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
@@ -74,7 +74,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;