[mlir] Allow Tile transform op to take dynamic sizes
authorAlex Zinenko <zinenko@google.com>
Thu, 7 Jul 2022 13:55:44 +0000 (15:55 +0200)
committerAlex Zinenko <zinenko@google.com>
Tue, 12 Jul 2022 12:21:54 +0000 (12:21 +0000)
commit4e4a4c057665d169b8fe6dcdd5bb7c0b0bf8ff19
tree522d380af2eb3da318fe5776a82c25d9fb332175
parent7b69843f0b502bb765206f534d7f50dcf88a60e5
[mlir] Allow Tile transform op to take dynamic sizes

Extend the definition of the Tile structured transform op to enable it
accepting handles to operations that produce tile sizes at runtime. This is
useful by itself and prepares for more advanced tiling strategies. Note that
the changes are relevant only to the transform dialect, the tiling
transformation itself already supports dynamic sizes.

Depends On D129216

Reviewed By: nicolasvasilache

Differential Revision: https://reviews.llvm.org/D129217
12 files changed:
mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.td
mlir/include/mlir/Dialect/Linalg/Utils/Utils.h
mlir/lib/Dialect/Linalg/TransformOps/LinalgTransformOps.cpp
mlir/lib/Dialect/Linalg/Transforms/Split.cpp
mlir/lib/Dialect/Linalg/Utils/Utils.cpp
mlir/python/mlir/dialects/_structured_transform_ops_ext.py
mlir/test/Dialect/Linalg/transform-op-scalarize.mlir
mlir/test/Dialect/Linalg/transform-op-tile.mlir
mlir/test/Dialect/Linalg/transform-ops.mlir
mlir/test/Dialect/Transform/selective-targeting.mlir
mlir/test/python/dialects/transform_structured_ext.py
utils/bazel/llvm-project-overlay/mlir/BUILD.bazel