diff options
author | MaheshRavishankar <1663364+MaheshRavishankar@users.noreply.github.com> | 2025-09-18 09:28:44 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2025-09-18 09:28:44 -0700 |
commit | b8649098a7fcf598406d8d8b7d68891d1444e9c8 (patch) | |
tree | fbd5039ac1458695c41d49780d9cbf8aa0429ded /llvm/lib/Support/CommandLine.cpp | |
parent | 4fabe6ffae885bddc52500ad59bc535febfaa494 (diff) | |
download | llvm-b8649098a7fcf598406d8d8b7d68891d1444e9c8.zip llvm-b8649098a7fcf598406d8d8b7d68891d1444e9c8.tar.gz llvm-b8649098a7fcf598406d8d8b7d68891d1444e9c8.tar.bz2 |
[mlir][SCF] Allow using a custom operation to generate loops with `mlir::tileUsingSCF`. (#159506)
This change adds an option to use a custom operation to generate the
inter-tile loops during tiling. When the loop type is set to
`scf::SCFTilingOptions::LoopType::CustomOp`, the method
`mlir::tileUsingSCF` provides two callback functions
1. First one to generate the header of the loop.
2. Second one to generate the terminator of the loop.
These methods receive the information needed to generate the
loops/terminator and expect to return information needed to generate
the code for the intra-tile computation. See comments for more
details.
Presently this is adds support only for tiling. Subsequent commits
will update this to add support for fusion as well.
The PR is split into two commits.
1) The first commit is an NFC that just refactors the code (and cleans
up some naming) to make it easier to add the support for custom loop
operations.
2) The second commit adds the support for using a custom loop operation,
as well as a test to exercise this path.
Signed-off-by: MaheshRavishankar <mahesh.ravishankar@gmail.com>
---------
Signed-off-by: MaheshRavishankar <mahesh.ravishankar@gmail.com>
Diffstat (limited to 'llvm/lib/Support/CommandLine.cpp')
0 files changed, 0 insertions, 0 deletions