[Mlir-commits] [mlir] [MLIR][Transform] Prefer entry points in current module (PR #151323)
Erick Ochoa Lopez
llvmlistbot at llvm.org
Thu Jul 31 07:10:37 PDT 2025
================
@@ -129,15 +129,29 @@ transform::detail::findTransformEntryPoint(Operation *root, ModuleOp module,
l.push_back(module);
for (Operation *op : l) {
transform::TransformOpInterface transform = nullptr;
- op->walk<WalkOrder::PreOrder>(
- [&](transform::NamedSequenceOp namedSequenceOp) {
+ for (Region ®ion : op->getRegions()) {
+ for (Block &block : region.getBlocks()) {
+ auto namedSequenceOps = block.getOps<transform::NamedSequenceOp>();
+ for (transform::NamedSequenceOp namedSequenceOp : namedSequenceOps) {
if (namedSequenceOp.getSymName() == entryPoint) {
transform = cast<transform::TransformOpInterface>(
namedSequenceOp.getOperation());
- return WalkResult::interrupt();
+ break;
}
- return WalkResult::advance();
- });
+ }
+ }
+ }
+ if (!transform) {
----------------
amd-eochoalo wrote:
https://github.com/llvm/llvm-project/pull/151323/commits/4cbdc979ca6b9a1d954ca29fc562864998693030 https://github.com/llvm/llvm-project/pull/151323/commits/a69dd77e968d2c477fdc2b0951fd0bf6a4383475 https://github.com/llvm/llvm-project/pull/151323/commits/fe8ee9b071fbbc1304070fb58a3df448f7a0d5b4
https://github.com/llvm/llvm-project/pull/151323
More information about the Mlir-commits
mailing list