Skip to content

Commit 96e778c

Browse files
author
Rolf Morel
committed
[MLIR][Python] Add missing peel_front argument to LoopPeelOp's extension class
While PR 74015 added support for peeling an iteration from the front of a loop, it did not update the (non-automatically generated) part of the Python interface that exposes the op. Some extraneous text to test https pushing.
1 parent b45de48 commit 96e778c

File tree

1 file changed

+6
-0
lines changed
  • mlir/python/mlir/dialects/transform

1 file changed

+6
-0
lines changed

mlir/python/mlir/dialects/transform/loop.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ def __init__(
5555
remainder_loop_type: Type,
5656
target: Union[Operation, Value],
5757
*,
58+
peel_front: Union[bool, BoolAttr] = False,
5859
fail_if_already_divisible: Union[bool, BoolAttr] = False,
5960
ip=None,
6061
loc=None,
@@ -63,6 +64,11 @@ def __init__(
6364
main_loop_type,
6465
remainder_loop_type,
6566
_get_op_result_or_value(target),
67+
peel_front=(
68+
peel_front
69+
if isinstance(peel_front, BoolAttr)
70+
else BoolAttr.get(peel_front)
71+
),
6672
fail_if_already_divisible=(
6773
fail_if_already_divisible
6874
if isinstance(fail_if_already_divisible, BoolAttr)

0 commit comments

Comments
 (0)