diff --git a/mlir/python/mlir/dialects/_structured_transform_ops_ext.py b/mlir/python/mlir/dialects/_structured_transform_ops_ext.py --- a/mlir/python/mlir/dialects/_structured_transform_ops_ext.py +++ b/mlir/python/mlir/dialects/_structured_transform_ops_ext.py @@ -110,6 +110,24 @@ ip=ip) +class MatchOp: + """Specialization for MatchOp class.""" + + @classmethod + def match_op_names(MatchOp, + target: Union[Operation, Value], + names: Sequence[str], + loc=None, + ip=None): + pdl_operation_type = pdl.OperationType.get() + return MatchOp( + pdl_operation_type, + _get_op_result_or_value(target), + ops=ArrayAttr.get(list(map(lambda s: StringAttr.get(s), names))), + loc=loc, + ip=ip) + + class MultiTileSizesOp: """Specialization for MultitileSizesOp class."""