[Mlir-commits] [mlir] 4911100 - [mlir][linalg][python] Add named constructor for MatchOp

[Matthias Springer]

Author: Matthias Springer
Date: 2022-08-15T14:41:28+02:00
New Revision: 4911100dc6c29271a54a147ca27fb9ba5522299f

URL: https://github.com/llvm/llvm-project/commit/4911100dc6c29271a54a147ca27fb9ba5522299f
DIFF: https://github.com/llvm/llvm-project/commit/4911100dc6c29271a54a147ca27fb9ba5522299f.diff

LOG: [mlir][linalg][python] Add named constructor for MatchOp

This constructor makes it easier to match for ops by their name.

Differential Revision: https://reviews.llvm.org/D131882

Added: 
    

Modified: 
    mlir/python/mlir/dialects/_structured_transform_ops_ext.py

Removed: 
    


################################################################################
diff  --git a/mlir/python/mlir/dialects/_structured_transform_ops_ext.py b/mlir/python/mlir/dialects/_structured_transform_ops_ext.py
index 95bf2cc992954..faf98ef49c306 100644
--- a/mlir/python/mlir/dialects/_structured_transform_ops_ext.py
+++ b/mlir/python/mlir/dialects/_structured_transform_ops_ext.py
@@ -110,6 +110,24 @@ def __init__(self,
         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."""