Fixes for llvm/llvm-project#143217

Signed-off-by: MaheshRavishankar <[email protected]>

Author: MaheshRavishankar

compiler/src/iree/compiler/Codegen/Common/CPU/CPUPrepareUkernels.cpp

@@ -75,7 +75,7 @@ static void tileNonPackedDimsFor3DPackOps(RewriterBase &rewriter,
     FailureOr<scf::SCFTilingResult> tilingResult =
         scf::tileUsingSCF(rewriter, tilingInterfaceOp, options);
     assert(succeeded(tilingResult));
-    rewriter.replaceOp(packOp, tilingResult->mergeResult.replacements);
+    rewriter.replaceOp(packOp, tilingResult->replacements);
   });
 }
 
@@ -110,7 +110,7 @@ static void tileNonPackedDimsFor5DPUnpackOps(RewriterBase &rewriter,
     FailureOr<scf::SCFTilingResult> tilingResult =
         scf::tileUsingSCF(rewriter, tilingInterfaceOp, options);
     assert(succeeded(tilingResult));
-    rewriter.replaceOp(unpackOp, tilingResult->mergeResult.replacements);
+    rewriter.replaceOp(unpackOp, tilingResult->replacements);
   });
 }
 

compiler/src/iree/compiler/Codegen/Common/DecomposePackUnPackOps.cpp

@@ -205,7 +205,7 @@ static LogicalResult commonRunOnOperation(
             unpackTilingOptions);
         if (failed(tilingResult))
           return WalkResult::interrupt();
-        rewriter.replaceOp(op, tilingResult->mergeResult.replacements);
+        rewriter.replaceOp(op, tilingResult->replacements);
         return WalkResult::advance();
       });
       if (status.wasInterrupted()) {

compiler/src/iree/compiler/Codegen/Common/GPU/GPUTensorTile.cpp

@@ -90,7 +90,7 @@ class TileConsumerAndFuseInputProducer final
     }
 
     // Replace the tiled op with replacements.
-    rewriter.replaceOp(op, tilingResult->mergeResult.replacements);
+    rewriter.replaceOp(op, tilingResult->replacements);
     filter.replaceLinalgTransformationFilter(rewriter,
                                              tilingResult->tiledOps.front());
 
@@ -291,7 +291,7 @@ static LogicalResult tileParallelDims(mlir::FunctionOpInterface funcOp,
     if (failed(tilingResult)) {
       return tilingOp->emitOpError("failed to tile to scf.forall");
     }
-    rewriter.replaceOp(tilingOp, tilingResult->mergeResult.replacements);
+    rewriter.replaceOp(tilingOp, tilingResult->replacements);
   }
   return success();
 }

compiler/src/iree/compiler/Codegen/Common/GPU/GPUTileReduction.cpp

@@ -38,11 +38,12 @@ static LogicalResult tileReduction(linalg::LinalgOp op) {
     sizes.push_back(rewriter.getIndexAttr(size));
   }
   rewriter.setInsertionPoint(op);
+
   FailureOr<scf::SCFTilingResult> results = scf::tileReductionUsingScf(
       rewriter, cast<PartialReductionOpInterface>(op.getOperation()), sizes);
   if (failed(results))
     return failure();
-  rewriter.replaceOp(op, results->mergeResult.replacements);
+  rewriter.replaceOp(op, results->replacements);
   return success();
 }
 

compiler/src/iree/compiler/Codegen/LLVMCPU/LLVMCPU2DScalableTo1DScalable.cpp

@@ -161,7 +161,7 @@ dropScalabilityFromUnsupportedOperations(mlir::FunctionOpInterface funcOp,
         setLoweringConfig(newOp, newLoweringConfig);
     }
 
-    rewriter.replaceOp(tilingOp, tilingResult->mergeResult.replacements);
+    rewriter.replaceOp(tilingOp, tilingResult->replacements);
   }
   return success();
 }

compiler/src/iree/compiler/Codegen/LLVMCPU/LLVMCPUSplitReduction.cpp

@@ -127,7 +127,7 @@ LogicalResult splitReductionImpl(Operation *op, int64_t size,
     LDBG("failed on step 1 (SCFTiling)");
     return failure();
   }
-  rewriter.replaceOp(linalgOp, tileResFirst->mergeResult.replacements);
+  rewriter.replaceOp(linalgOp, tileResFirst->replacements);
 
   // 2) Apply splitReduction on the single vector-length array.
   // splitReduction already replaces the op.
@@ -154,8 +154,7 @@ LogicalResult splitReductionImpl(Operation *op, int64_t size,
     LDBG("failed on step 3 (SCFTiling)");
     return failure();
   }
-  rewriter.replaceOp(splitRes->splitLinalgOp,
-                     tileRes->mergeResult.replacements);
+  rewriter.replaceOp(splitRes->splitLinalgOp, tileRes->replacements);
   return success();
 }
 

compiler/src/iree/compiler/Codegen/LLVMCPU/LLVMCPUTile.cpp

@@ -102,7 +102,7 @@ void LLVMCPUTilePass::runOnOperation() {
         scf::tileUsingSCF(rewriter, op, options);
     if (failed(tiledResults))
       continue;
-    rewriter.replaceOp(op, tiledResults->mergeResult.replacements);
+    rewriter.replaceOp(op, tiledResults->replacements);
   }
 
   RewritePatternSet patterns =

compiler/src/iree/compiler/Codegen/Transforms/Transforms.cpp

@@ -1627,7 +1627,7 @@ LogicalResult tileLinalgOpsWithFilter(mlir::FunctionOpInterface funcOp,
     for (auto tiledOp : tiledResults->tiledOps) {
       filter.replaceLinalgTransformationFilter(rewriter, tiledOp);
     }
-    rewriter.replaceOp(op, tiledResults->mergeResult.replacements);
+    rewriter.replaceOp(op, tiledResults->replacements);
   }
 
   return success();