Bump MLIR bindings

lighthouse/dialects/transform/smt_ext/ops/constrain_params.py

@@ -78,11 +78,13 @@ def allow_repeated_handle_operands(_op: "ConstrainParamsOp") -> bool:
 
     class ConstrainParamsMemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: "ConstrainParamsOp", effects):
+        def get_effects(op: "ConstrainParamsOp"):
+            effects = []
             if op.op_operands:
-                transform.only_reads_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.only_reads_payload(effects)
+                effects += transform.only_reads_handle(op.op_operands)
+            effects += transform.produces_handle(op.results)
+            effects += transform.only_reads_payload()
+            return effects
 
 
 class MixedResultConstrainParamsOp(ConstrainParamsOp):

lighthouse/dialects/transform/transform_ext/ops/assign_tile_sizes.py

@@ -73,10 +73,12 @@ def allow_repeated_handle_operands(_op: "AssignTileSizesOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: ir.Operation, effects):
-            transform.only_reads_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.modifies_payload(effects)
+        def get_effects(op: ir.Operation):
+            return (
+                transform.only_reads_handle(op.op_operands)
+                + transform.produces_handle(op.results)
+                + transform.modifies_payload()
+            )
 
 
 def assign_tile_sizes(

lighthouse/dialects/transform/transform_ext/ops/clear_tile_and_fuse_annotations.py

@@ -66,10 +66,12 @@ def allow_repeated_handle_operands(
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: ir.Operation, effects):
-            transform.only_reads_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.modifies_payload(effects)
+        def get_effects(op: ir.Operation):
+            return (
+                transform.only_reads_handle(op.op_operands)
+                + transform.produces_handle(op.results)
+                + transform.modifies_payload()
+            )
 
 
 def clear_tile_and_fuse_annotations(

lighthouse/dialects/transform/transform_ext/ops/convert_func_results_to_args.py

@@ -128,10 +128,12 @@ def allow_repeated_handle_operands(_op: "ConvertFuncResultsToArgsOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: "ConvertFuncResultsToArgsOp", effects):
-            transform.consumes_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.modifies_payload(effects)
+        def get_effects(op: "ConvertFuncResultsToArgsOp"):
+            return (
+                transform.consumes_handle(op.op_operands)
+                + transform.produces_handle(op.results)
+                + transform.modifies_payload()
+            )
 
 
 def convert_func_results_to_args(

lighthouse/dialects/transform/transform_ext/ops/extract_handle.py

@@ -55,10 +55,12 @@ def allow_repeated_handle_operands(_op: "ExtractHandleOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: ir.Operation, effects):
-            transform.only_reads_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.only_reads_payload(effects)
+        def get_effects(op: ir.Operation):
+            return (
+                transform.only_reads_handle(op.op_operands)
+                + transform.produces_handle(op.results)
+                + transform.only_reads_payload()
+            )
 
 
 def extract_handle(

lighthouse/dialects/transform/transform_ext/ops/fold_singleton_extract_slice.py

@@ -184,10 +184,12 @@ def allow_repeated_handle_operands(_op: "FoldSingletonExtractSliceOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: ir.Operation, effects):
-            transform.only_reads_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.modifies_payload(effects)
+        def get_effects(op: ir.Operation):
+            return (
+                transform.only_reads_handle(op.op_operands)
+                + transform.produces_handle(op.results)
+                + transform.modifies_payload()
+            )
 
 
 def fold_singleton_extract_slice(

lighthouse/dialects/transform/transform_ext/ops/get_fusion_roots.py

@@ -149,10 +149,12 @@ def allow_repeated_handle_operands(_op: "GetFusionRootsOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: ir.Operation, effects):
-            transform.only_reads_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.only_reads_payload(effects)
+        def get_effects(op: ir.Operation):
+            return (
+                transform.only_reads_handle(op.op_operands)
+                + transform.produces_handle(op.results)
+                + transform.only_reads_payload()
+            )
 
 
 def get_fusion_roots(

lighthouse/dialects/transform/transform_ext/ops/get_leading_unit_tile_sizes.py

@@ -57,10 +57,12 @@ def allow_repeated_handle_operands(_op: "GetLeadingUnitTileSizesOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: ir.Operation, effects):
-            transform.only_reads_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.only_reads_payload(effects)
+        def get_effects(op: ir.Operation):
+            return (
+                transform.only_reads_handle(op.op_operands)
+                + transform.produces_handle(op.results)
+                + transform.only_reads_payload()
+            )
 
 
 def get_leading_unit_tile_sizes(

lighthouse/dialects/transform/transform_ext/ops/get_named_attribute.py

@@ -49,10 +49,12 @@ def allow_repeated_handle_operands(_op: "GetNamedAttributeOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: ir.Operation, effects):
-            transform.only_reads_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.only_reads_payload(effects)
+        def get_effects(op: ir.Operation):
+            return (
+                transform.only_reads_handle(op.op_operands)
+                + transform.produces_handle(op.results)
+                + transform.only_reads_payload()
+            )
 
 
 def get_named_attribute(

lighthouse/dialects/transform/transform_ext/ops/get_tile_sizes.py

@@ -59,10 +59,12 @@ def allow_repeated_handle_operands(_op: "GetTileSizesOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: ir.Operation, effects):
-            transform.only_reads_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.only_reads_payload(effects)
+        def get_effects(op: ir.Operation):
+            return (
+                transform.only_reads_handle(op.op_operands)
+                + transform.produces_handle(op.results)
+                + transform.only_reads_payload()
+            )
 
 
 def get_tile_sizes(

lighthouse/dialects/transform/transform_ext/ops/get_tileable_consumers.py

@@ -91,10 +91,12 @@ def allow_repeated_handle_operands(_op: "GetTileableConsumersOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: ir.Operation, effects):
-            transform.only_reads_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.only_reads_payload(effects)
+        def get_effects(op: ir.Operation):
+            return (
+                transform.only_reads_handle(op.op_operands)
+                + transform.produces_handle(op.results)
+                + transform.only_reads_payload()
+            )
 
 
 def get_tileable_consumers(

lighthouse/dialects/transform/transform_ext/ops/get_tiling_sizes.py

@@ -146,10 +146,12 @@ def allow_repeated_handle_operands(_op: "GetTilingSizesOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: ir.Operation, effects):
-            transform.only_reads_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.only_reads_payload(effects)
+        def get_effects(op: ir.Operation):
+            return (
+                transform.only_reads_handle(op.op_operands)
+                + transform.produces_handle(op.results)
+                + transform.only_reads_payload()
+            )
 
 
 def get_tiling_sizes(

lighthouse/dialects/transform/transform_ext/ops/move_offsets_to_subview.py

@@ -168,10 +168,12 @@ def allow_repeated_handle_operands(_op: "MoveOffsetsToSubviewOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: "MoveOffsetsToSubviewOp", effects):
-            transform.consumes_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.modifies_payload(effects)
+        def get_effects(op: "MoveOffsetsToSubviewOp"):
+            return (
+                transform.consumes_handle(op.op_operands)
+                + transform.produces_handle(op.results)
+                + transform.modifies_payload()
+            )
 
 
 def move_offsets_to_subview(

lighthouse/dialects/transform/transform_ext/ops/param_cmp_eq.py

@@ -44,9 +44,11 @@ def allow_repeated_handle_operands(_op: "ParamCmpEqOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: "ParamCmpEqOp", effects):
-            transform.only_reads_handle(op.op_operands, effects)
-            transform.only_reads_payload(effects)
+        def get_effects(op: "ParamCmpEqOp"):
+            return (
+                transform.only_reads_handle(op.op_operands)
+                + transform.only_reads_payload()
+            )
 
 
 def param_cmp_eq(lhs: ir.Value, rhs: ir.Value):

lighthouse/dialects/transform/transform_ext/ops/propagate_tile_sizes.py

@@ -135,10 +135,12 @@ def allow_repeated_handle_operands(_op: "PropagateTileSizesOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: ir.Operation, effects):
-            transform.only_reads_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.modifies_payload(effects)
+        def get_effects(op: ir.Operation):
+            return (
+                transform.only_reads_handle(op.op_operands)
+                + transform.produces_handle(op.results)
+                + transform.modifies_payload()
+            )
 
 
 def propagate_tile_sizes(

lighthouse/dialects/transform/transform_ext/ops/replace.py

@@ -101,12 +101,15 @@ def allow_repeated_handle_operands(_op: "ReplaceOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: ir.Operation, effects):
-            transform.consumes_handle(op.op_operands[:1], effects)
+        def get_effects(op: ir.Operation):
+            effects = transform.consumes_handle(op.op_operands[:1])
             if new_operands_handles := op.op_operands[1:]:
-                transform.only_reads_handle(new_operands_handles, effects)
-            transform.produces_handle(op.results, effects)
-            transform.modifies_payload(effects)
+                effects += transform.only_reads_handle(new_operands_handles)
+            return (
+                effects
+                + transform.produces_handle(op.results)
+                + transform.modifies_payload()
+            )
 
 
 def replace(

lighthouse/dialects/transform/transform_ext/ops/replace_with_fused_attention.py

@@ -470,15 +470,17 @@ def allow_repeated_handle_operands(_op: "ReplaceWithFusedAttentionOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: ir.Operation, effects):
-            # Read Q, K, scale, V slices
-            transform.only_reads_handle(op.op_operands[:4], effects)
-            # Consume and replace output
-            transform.consumes_handle(op.op_operands[4:5], effects)
-            # Produce new output handle
-            transform.produces_handle(op.results, effects)
-            # Modify the payload
-            transform.modifies_payload(effects)
+        def get_effects(op: ir.Operation):
+            return (
+                # Read Q, K, scale, V slices
+                transform.only_reads_handle(op.op_operands[:4])
+                # Consume and replace output
+                + transform.consumes_handle(op.op_operands[4:5])
+                # Produce new output handle
+                + transform.produces_handle(op.results)
+                # Modify the payload
+                + transform.modifies_payload()
+            )
 
 
 def replace_with_fused_attention(

lighthouse/dialects/transform/transform_ext/ops/reverse_handles.py

@@ -41,10 +41,12 @@ def allow_repeated_handle_operands(_op: "ReverseHandlesOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: ir.Operation, effects):
-            transform.only_reads_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.only_reads_payload(effects)
+        def get_effects(op: ir.Operation):
+            return (
+                transform.only_reads_handle(op.op_operands)
+                + transform.produces_handle(op.results)
+                + transform.only_reads_payload()
+            )
 
 
 def reverse_handles(

lighthouse/dialects/transform/transform_ext/ops/trace_producers.py

@@ -69,10 +69,12 @@ def allow_repeated_handle_operands(_op: "TraceProducersOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: ir.Operation, effects):
-            transform.only_reads_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.only_reads_payload(effects)
+        def get_effects(op: ir.Operation):
+            return (
+                transform.only_reads_handle(op.op_operands)
+                + transform.produces_handle(op.results)
+                + transform.only_reads_payload()
+            )
 
 
 def trace_producers(

lighthouse/dialects/transform/transform_ext/ops/update_address_space.py

@@ -87,10 +87,12 @@ def allow_repeated_handle_operands(_op: "UpdateAddressSpaceOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: ir.Operation, effects):
-            transform.consumes_handle(op.op_operands[:1], effects)
-            transform.produces_handle(op.results, effects)
-            transform.modifies_payload(effects)
+        def get_effects(op: ir.Operation):
+            return (
+                transform.consumes_handle(op.op_operands[:1])
+                + transform.produces_handle(op.results)
+                + transform.modifies_payload()
+            )
 
 
 def update_address_space(

lighthouse/dialects/transform/transform_ext/ops/wrap_in_benching_func.py

@@ -101,10 +101,12 @@ def allow_repeated_handle_operands(_op: "WrapInBenchingFuncOp") -> bool:
 
     class MemoryEffectsOpInterfaceModel(ir.MemoryEffectsOpInterface):
         @staticmethod
-        def get_effects(op: "WrapInBenchingFuncOp", effects):
-            transform.only_reads_handle(op.op_operands, effects)
-            transform.produces_handle(op.results, effects)
-            transform.modifies_payload(effects)
+        def get_effects(op: "WrapInBenchingFuncOp"):
+            return (
+                transform.only_reads_handle(op.op_operands)
+                + transform.produces_handle(op.results)
+                + transform.modifies_payload()
+            )
 
 
 def wrap_in_benching_func(