From 6ac24ce6569d38049d03e8fa434242c3b336c888 Mon Sep 17 00:00:00 2001 From: Alex Zinenko Date: Fri, 20 Jan 2023 16:02:31 +0000 Subject: [PATCH] [mlir] fix python types --- mlir/python/mlir/dialects/_structured_transform_ops_ext.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mlir/python/mlir/dialects/_structured_transform_ops_ext.py b/mlir/python/mlir/dialects/_structured_transform_ops_ext.py index f045e5c..97705e2 100644 --- a/mlir/python/mlir/dialects/_structured_transform_ops_ext.py +++ b/mlir/python/mlir/dialects/_structured_transform_ops_ext.py @@ -254,7 +254,7 @@ class TileOp: @overload def __init__(self, - target: Union[Operation, Value], + target: Union[Operation, Value, OpView], *, sizes: Optional[Union[Sequence[Union[int, IntegerAttr, Operation, Value]], ArrayAttr]] = None, @@ -265,7 +265,7 @@ class TileOp: def __init__(self, loop_types_or_target: Union[Type, List[Type], Operation, Value], - target_or_none: Optional[Union[Operation, Value]] = None, + target_or_none: Optional[Union[Operation, Value, OpView]] = None, *, sizes: Optional[Union[Sequence[Union[int, IntegerAttr, Operation, Value]], ArrayAttr]] = None, @@ -291,7 +291,7 @@ class TileOp: num_loops = sum( v if v == 0 else 1 for v in self.__extract_values(sizes_attr)) - if isinstance(loop_types_or_target, (Operation, Value)): + if isinstance(loop_types_or_target, (Operation, Value, OpView)): loop_types = [transform.AnyOpType.get()] * num_loops target = loop_types_or_target assert target_or_none is None, "Cannot construct TileOp with two targets." -- 2.7.4