Skip to content

Commit 9ba3723

Browse files
authored
fix region_adder (#48)
1 parent d270ac6 commit 9ba3723

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

mlir/extras/util.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -325,6 +325,8 @@ def bb(*preds: Tuple[Union[Successor, OpView]]):
325325
def region_adder(terminator=None):
326326
def wrapper(op_region_adder):
327327
def region_adder_decorator(op, *args, **kwargs):
328+
if isinstance(op, Value):
329+
op = op.owner.opview
328330
region = op_region_adder(op, *args, **kwargs)
329331

330332
return op_region_builder(op, region, terminator)

0 commit comments

Comments
 (0)