diff options
| author | Gaurav Shukla <gaurav@nod-labs.com> | 2024-04-21 17:07:02 +0530 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-04-21 07:37:02 -0400 |
| commit | e095d978ba476c9624b4e72776089ea7301fa657 (patch) | |
| tree | 699b281d94939ba3cd6612f5193cc4b0e6e558bc /utils | |
| parent | aa22d4422ee031d3867290e6ec12985f87d9ea2f (diff) | |
| download | llvm-e095d978ba476c9624b4e72776089ea7301fa657.zip llvm-e095d978ba476c9624b4e72776089ea7301fa657.tar.gz llvm-e095d978ba476c9624b4e72776089ea7301fa657.tar.bz2 | |
[MLIR] Generalize expand_shape to take shape as explicit input (#69267)
This patch generalizes tensor.expand_shape and memref.expand_shape to
consume the output shape as a list of SSA values. This enables us to
implement generic reshape operations with dynamic shapes using
collapse_shape/expand_shape pairs.
The output_shape input to expand_shape follows the static/dynamic
representation that's also used in `tensor.extract_slice`.
Differential Revision: https://reviews.llvm.org/D140821
Co-authored-by: Ramiro Leal-Cavazos <ramiroleal050@gmail.com>
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/bazel/llvm-project-overlay/mlir/BUILD.bazel | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel index 6c732b8..8c6752e 100644 --- a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel +++ b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel @@ -3831,6 +3831,7 @@ cc_library( includes = ["include"], deps = [ ":DialectUtilsIncGen", + ":ArithDialect", ":IR", ":Support", "//llvm:Support", |