Fix problem where the shape of the insert shape was calculated incorrectly

Author: josel-amd

mlir/lib/Dialect/Linalg/Transforms/TilingInterfaceImpl.cpp

@@ -218,10 +218,15 @@ struct LinalgOpTilingInterface
         }));
 
     OpOperand *outOperand = linalgOp.getDpsInitOperand(resultNumber);
+    SmallVector<OpFoldResult> allShapeSizes =
+        linalgOp.createFlatListOfOperandDims(b, linalgOp.getLoc());
+    SmallVector<OpFoldResult> sizeBounds =
+        mlir::affine::makeComposedFoldedMultiResultAffineApply(
+            b, loc, linalgOp.getShapesToLoopsMap(), allShapeSizes);
     SliceParameters sliceParams = computeSliceParameters(
         b, loc, outOperand->get(), sizes,
         linalgOp.getMatchingIndexingMap(outOperand), offsets,
-        /*ubs*/ {}, subShapeSizes, true);
+        /*ubs*/ sizeBounds, subShapeSizes, true);
     resultOffsets = sliceParams.offsets;
     resultSizes = sliceParams.sizes;
     return success();

mlir/lib/Dialect/SCF/Transforms/TileUsingInterface.cpp

@@ -1905,9 +1905,10 @@ mlir::scf::tileAndFuseConsumerOfSlice(RewriterBase &rewriter,
     SmallVector<SmallVector<OpFoldResult>> resultSizes(
         totalNumResultsOfConsumer);
     for (auto [idx, v] : llvm::enumerate(tiledConsumerOp->getResults())) {
-      if (failed(tiledConsumerOp.getResultTilePosition(
-              rewriter, idx, iterDomainOffsets, iterDomainSizes,
-              resultOffsets[idx], resultSizes[idx]))) {
+      if (failed(cast<TilingInterface>(clonedConsumerOp)
+                     .getResultTilePosition(rewriter, idx, iterDomainOffsets,
+                                            iterDomainSizes, resultOffsets[idx],
+                                            resultSizes[idx]))) {
         return rewriter.notifyMatchFailure(
             tiledConsumerOp,
             "can't get result domain position from iter domain position");