diff options
| author | Nicolas Vasilache <nicolas.vasilache@gmail.com> | 2023-03-21 14:41:20 -0700 |
|---|---|---|
| committer | Nicolas Vasilache <nicolas.vasilache@gmail.com> | 2023-03-23 04:03:27 -0700 |
| commit | 4dc72d47ce88218ff3c6a7ae724beb6ab3ba2ade (patch) | |
| tree | 2815c3114d7dafb5953e965d1d05a5ca8d3ad5b3 /mlir/lib/Dialect/Tensor/Utils/Utils.cpp | |
| parent | 002c4b7b955b1fc8825b4d6b46bb079390bce812 (diff) | |
[mlir][Tensor] Add a FoldTensorSubsetOps pass and patterns
These patterns follow FoldMemRefAliasOps which is further refactored for reuse.
In the process, fix FoldMemRefAliasOps handling of strides for vector.transfer ops which was previously incorrect.
These opt-in patterns generalize the existing canonicalizations on vector.transfer ops.
In the future the blanket canonicalizations will be retired.
They are kept for now to minimize porting disruptions.
Differential Revision: https://reviews.llvm.org/D146624
Diffstat (limited to 'mlir/lib/Dialect/Tensor/Utils/Utils.cpp')
| -rw-r--r-- | mlir/lib/Dialect/Tensor/Utils/Utils.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/mlir/lib/Dialect/Tensor/Utils/Utils.cpp b/mlir/lib/Dialect/Tensor/Utils/Utils.cpp index a5847250fa91..4c09c540bde2 100644 --- a/mlir/lib/Dialect/Tensor/Utils/Utils.cpp +++ b/mlir/lib/Dialect/Tensor/Utils/Utils.cpp @@ -14,6 +14,7 @@ #include "mlir/Dialect/Affine/IR/AffineOps.h" #include "mlir/Dialect/Arith/IR/Arith.h" +#include "mlir/Dialect/Arith/Utils/Utils.h" #include "mlir/Dialect/Utils/IndexingUtils.h" using namespace mlir; |