diff options
| author | Gaurav Shukla <gaurav@nod-labs.com> | 2024-04-30 21:58:35 +0530 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-04-30 09:28:35 -0700 |
| commit | 97069a86193a617a9e4cf742a29db6116b2bf449 (patch) | |
| tree | 25550fd965bb34baea28efe74730180a0b4e994e /utils | |
| parent | 539f626ecd0c5148228fb6d2eda6228b4f2b2fca (diff) | |
| download | llvm-97069a86193a617a9e4cf742a29db6116b2bf449.zip llvm-97069a86193a617a9e4cf742a29db6116b2bf449.tar.gz llvm-97069a86193a617a9e4cf742a29db6116b2bf449.tar.bz2 | |
[MLIR] Generalize expand_shape to take shape as explicit input (#90040)
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
---------
Signed-off-by: Gaurav Shukla<gaurav.shukla@amd.com>
Signed-off-by: Gaurav Shukla <gaurav.shukla@amd.com>
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 52c874c..acd2d3a 100644 --- a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel +++ b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel @@ -12635,6 +12635,7 @@ cc_library( deps = [ ":ArithDialect", ":ComplexDialect", + ":DialectUtils", ":IR", "//llvm:Support", ], |