intel/llvm
1
0
Fork 0
mirror of https://github.com/intel/llvm.git synced 2026-02-05 22:17:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

[mlir][transform][structured][python] Allow str arg in match_op_names.

Browse Source 
Allow the `names` argument in `MatchOp.match_op_names` to be of type
`str` in addition to `Sequence[str]`. In this case, the argument is
treated as a list with one name, i.e., it is possible to write
`MatchOp.match_op_names(..., "test.dummy")` instead of
`MatchOp.match_op_names(..., ["test.dummy"])`.

Reviewed By: ftynse

Differential Revision: https://reviews.llvm.org/D155807
This commit is contained in:
Ingo Müller
2023-07-20 09:58:41 +00:00
parent 522831384f
commit 8fd207fd0d
2 changed files with 23 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
mlir/python/mlir/dialects/_structured_transform_ops_ext.py
View File

@@ -195,7 +195,7 @@ class MatchOp:
def match_op_names(
cls,
target: Union[Operation, Value],
names: Sequence[str],
names: Union[str, Sequence[str]],
*,
loc=None,
ip=None,
@@ -208,7 +208,7 @@ class MatchOp:
cls,
result_type: Type,
target: Union[Operation, Value],
names: Sequence[str],
names: Union[str, Sequence[str]],
*,
loc=None,
ip=None,
@@ -219,8 +219,8 @@ class MatchOp:
def match_op_names(
cls,
result_type_or_target: Union[Type, Operation, Value],
target_or_names: Union[Operation, Value, Sequence[str]],
names_or_none: Optional[Sequence[str]] = None,
target_or_names: Union[Operation, Value, Sequence[str], str],
names_or_none: Optional[Union[Sequence[str], str]] = None,
*,
loc=None,
ip=None,
@@ -234,6 +234,9 @@ class MatchOp:
target = result_type_or_target
names = target_or_names
if isinstance(names, str):
names = [names]
return cls(
result_type,
_get_op_result_or_value(target),