[mlir][linalg] Add extra parameter to tiling reduction to foreach_thread
authorThomas Raoux <thomasraoux@google.com>
Tue, 6 Dec 2022 23:14:00 +0000 (23:14 +0000)
committerThomas Raoux <thomasraoux@google.com>
Wed, 7 Dec 2022 18:37:05 +0000 (18:37 +0000)
commitf7fda6ba4a7bca12bc7ef62d27895aa24482eda5
tree71187f4028eb9ccf6a9993aa356146c47eeb8abd
parent2c41310fc146a1f609147c65ac5f30e5a57e84a8
[mlir][linalg] Add extra parameter to tiling reduction to foreach_thread

This adds a tile_size parameter, when it is used the tiles are
cyclically distributed onto the threads of the scf.foreach_thread op.

Differential Revision: https://reviews.llvm.org/D139474
mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.td
mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
mlir/lib/Dialect/Linalg/TransformOps/LinalgTransformOps.cpp
mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp
mlir/test/Dialect/Linalg/transform-tile-reduction.mlir