Propagate reshapes through generics with reduction iterators

Signed-off-by: Ian Wood <ianwood2024@u.northwestern.edu>

compiler/src/iree/compiler/DispatchCreation/BubbleUpExpandShapes.cpp

@@ -60,9 +60,7 @@ void BubbleUpExpandShapesPass::runOnOperation() {
         // Do not fuse producer generic op if it has more than one user
         // or any reduction iterators.
         if (auto producerGenericOp = dyn_cast<linalg::GenericOp>(producer)) {
-          return producerGenericOp->hasOneUse() &&
-                 llvm::all_of(producerGenericOp.getIteratorTypesArray(),
-                              linalg::isParallelIterator);
+          return producerGenericOp->hasOneUse();
         }
 
         // Do not fuse with any producer linalg named ops for now.
@@ -72,9 +70,7 @@ void BubbleUpExpandShapesPass::runOnOperation() {
 
         // Do not fuse with consumer linalg named ops or reductions.
         if (auto consumerLinalgOp = dyn_cast<linalg::LinalgOp>(consumer)) {
-          return isa<linalg::GenericOp>(consumerLinalgOp) &&
-                 llvm::all_of(consumerLinalgOp.getIteratorTypesArray(),
-                              linalg::isParallelIterator);
+          return isa<linalg::GenericOp>(consumerLinalgOp);
         }
         // Fuse in all other cases.
         return true;