[Mlir-commits] [mlir] [MLIR][Transform] Prefer entry points in current module (PR #151323)

[Erick Ochoa Lopez]

================
@@ -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 &region : 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