[mlir][linalg] Fix result type in FoldSourceTensorCast
authorMatthias Springer <springerm@google.com>
Fri, 24 Sep 2021 07:39:37 +0000 (16:39 +0900)
committerMatthias Springer <springerm@google.com>
Fri, 24 Sep 2021 07:47:18 +0000 (16:47 +0900)
commitf3f25ffc04c0cbcc9a9bfc1b32b61750e8934ea8
tree42eba0e88a5464050ab3d55de7086ab06af53ed1
parent03142c5f67788bcc1573f76732d0fccd75c6b965
[mlir][linalg] Fix result type in FoldSourceTensorCast

* Do not discard static result type information that cannot be inferred from lower/upper padding.
* Add optional argument to `PadTensorOp::inferResultType` for specifying known result dimensions.

Differential Revision: https://reviews.llvm.org/D110380
mlir/include/mlir/Dialect/Linalg/IR/LinalgOps.td
mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp
mlir/test/Dialect/Linalg/canonicalize.mlir