fix region_adder (#48)

makslevental · Jan 16, 2024 · 9ba3723 · 9ba3723
1 parent d270ac6
commit 9ba3723
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mlir/extras/util.py b/mlir/extras/util.py
@@ -325,6 +325,8 @@ def bb(*preds: Tuple[Union[Successor, OpView]]):
 def region_adder(terminator=None):
     def wrapper(op_region_adder):
         def region_adder_decorator(op, *args, **kwargs):
+            if isinstance(op, Value):
+                op = op.owner.opview
             region = op_region_adder(op, *args, **kwargs)
 
             return op_region_builder(op, region, terminator)