diff options
author | Alex Zinenko <zinenko@google.com> | 2023-01-20 16:02:31 +0000 |
---|---|---|
committer | Alex Zinenko <zinenko@google.com> | 2023-01-20 16:03:27 +0000 |
commit | 6ac24ce6569d38049d03e8fa434242c3b336c888 (patch) | |
tree | b86bfff742e55cf45f5883062bdba30a7635ab23 /mlir/python | |
parent | 99c12afeb4a1afa608d487351eebe79ebf058b8d (diff) | |
download | llvm-6ac24ce6569d38049d03e8fa434242c3b336c888.tar.gz |
[mlir] fix python types
Diffstat (limited to 'mlir/python')
-rw-r--r-- | mlir/python/mlir/dialects/_structured_transform_ops_ext.py | 6 |
1 files 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 f045e5c13c1e..97705e2ad652 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." |