[mlir][linalg] Use getAsOpFoldResult in padding (NFC).

[gysit]

Summary

After padding, we introduce a ExtractSliceOp to get the final unpadded result. This revision uses getAsOpFoldResult to compute the size of the unpadded result, which guarantees the result type has a partially static shape if some of the sizes of the unpadded result are statically known. At the moment, we rely on canonicalization to cleanup the types after padding.

Depends On D114085

[gysit]

Rebase.

[gysit]

Rebase.

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

linalg::rewriteAsPaddedOp(OpBuilder &b, LinalgOp opToPad,
   paddedSubviewResults.reserve(opToPad->getNumResults());
   for (auto en : llvm::enumerate(paddedOp->getResults())) {
     Value paddedResult = en.value();
     int64_t resultNumber = en.index();
     int64_t rank = paddedResult.getType().cast<RankedTensorType>().getRank();
     SmallVector<OpFoldResult> offsets(rank, b.getIndexAttr(0));
     SmallVector<OpFoldResult> sizes;
     for (Value v : reifiedResultShapes[resultNumber])
-      sizes.push_back(v);
+      sizes.push_back(getAsOpFoldResult(v));
     SmallVector<OpFoldResult> strides(rank, b.getIndexAttr(1));
     paddedSubviewResults.push_back(b.create<tensor::ExtractSliceOp>(
         loc, paddedResult, offsets, sizes, strides));
   }
   return paddedSubviewResults;
 }
 
 /// Linalg base tiling pattern.