[Mlir-commits] [mlir] [MLIR] Add pattern to bubble up tensor.extract_slice (PR #126898)

[Quinn Dawkins]

================
@@ -111,6 +111,16 @@ def ApplyReassociativeReshapeFoldingPatternsOp : Op<Transform_Dialect,
   let assemblyFormat = "attr-dict";
 }
 
+def ApplyBubbleUpExtractSlicePatternsOp : Op<Transform_Dialect,
+    "apply_patterns.tensor.bubble_up_extract_slice",
+    [DeclareOpInterfaceMethods<PatternDescriptorOpInterface>]> {
+  let description = [{
+    Indicates that tensor.extract_slice and its producer should swap location.
----------------
qedawkins wrote:

nit: `swap location` reads as imprecise here. Something like
```suggestion
    Indicates that producers of tensor.extract_slice should swap and operate on the result of the slice.
```

https://github.com/llvm/llvm-project/pull/126898