Google Git
Sign in
llvm / llvm-project / mlir / ba4e03e4144915ddf10f0b5f7fec65b1674bd6b8
commitba4e03e4144915ddf10f0b5f7fec65b1674bd6b8[log] [tgz]
authorBangtian Liu <liubangtian@gmail.com>Wed Sep 10 15:49:24 2025 -0400
committerCopybara-Service <copybara-worker@google.com>Wed Sep 10 12:50:43 2025 -0700
treef1c5f792c6e3c5b0d2c3a6ad17cd22b08d4918ff
parent2cf6cafd94caa95f4928d4418756be3725c0e7d9 [diff]
[mlir][Linalg] Allow PartialReductionOpInterface ops in tile_reduction_using_forall (#157932)

Following [PR
#120118](https://github.com/llvm/llvm-project/pull/120118), this PR
extends transform.structured.tile_reduction_using_forall so that it can
be applied to any operation implementing `PartialReductionOpInterface`,
rather than being restricted to LinalgOp.

Existing tests relevant to linalg ops remain valid:
https://github.com/llvm/llvm-project/blob/2a2296b1aab4614bf6c95c3003000832c9d43de5/mlir/test/Dialect/Linalg/transform-tile-reduction.mlir#L114

Additional tests for non-Linalg operations (e.g., IREE custom ops that
implement `PartialReductionOpInterface`) will be added on the IREE side.

Signed-off-by: Bangtian Liu <liubangtian@gmail.com>
GitOrigin-RevId: 17723e472e228be5404ab4377498b52a0c5db03b
2 files changed
tree: f1c5f792c6e3c5b0d2c3a6ad17cd22b08d4918ff
  1. benchmark/
  2. cmake/
  3. docs/
  4. examples/
  5. include/
  6. lib/
  7. python/
  8. test/
  9. tools/
  10. unittests/
  11. utils/
  12. .clang-format
  13. .clang-tidy
  14. CMakeLists.txt
  15. LICENSE.TXT
  16. Maintainers.md
  17. README.md
README.md

Multi-Level Intermediate Representation

See https://mlir.llvm.org/ for more information.