[Mlir-commits] [mlir] [MLIR] [python] Fixed a bug in `PyRegionList` iterator (PR #137232)

Maksim Levental llvmlistbot at llvm.org
Thu Apr 24 17:01:39 PDT 2025


================
@@ -82,12 +82,16 @@ def walk_operations(indent, op):
     # CHECK:           OP 1: func.return
     walk_operations("", op)
 
-    # CHECK:    Region iter: <mlir.{{.+}}.RegionIterator
-    # CHECK:     Block iter: <mlir.{{.+}}.BlockIterator
-    # CHECK: Operation iter: <mlir.{{.+}}.OperationIterator
-    print("   Region iter:", iter(op.regions))
-    print("    Block iter:", iter(op.regions[-1]))
-    print("Operation iter:", iter(op.regions[-1].blocks[-1]))
+    # CHECK:    Regions: [<mlir.{{.+}}.Region object at {{.+}}>]
+    # CHECK:     Blocks: [<mlir.{{.+}}.Block object at {{.+}}>]
+    # CHECK: Operations: [<mlir.{{.+}}.FuncOp object at {{.+}}>]
+    print("   Regions:", [*op.regions])
+    print("    Blocks:", [*op.regions[-1]])
+    print("Operations:", [*op.regions[-1].blocks[-1]])
----------------
makslevental wrote:

i get this fine but others might not (a little too "pythonic"); prefer

```suggestion
    print("   Regions:", list(op.regions))
    print("    Blocks:", list(op.regions[-1]))
    print("Operations:", list(op.regions[-1].blocks[-1]))
```

https://github.com/llvm/llvm-project/pull/137232


More information about the Mlir-commits mailing list