diff --git a/mlir/include/mlir/Dialect/Linalg/Passes.td b/mlir/include/mlir/Dialect/Linalg/Passes.td index 44da2965e6892..42d4d2083fc1c 100644 --- a/mlir/include/mlir/Dialect/Linalg/Passes.td +++ b/mlir/include/mlir/Dialect/Linalg/Passes.td @@ -167,8 +167,15 @@ def LinalgInlineScalarOperandsPass : Pass<"linalg-inline-scalar-operands"> { } def LinalgFoldIntoElementwisePass : Pass<"linalg-fold-into-elementwise"> { - let summary = "Fold transform, broadcast and other ops into elementwise"; + let summary = "Fold transpose ops into elementwise"; let dependentDialects = ["linalg::LinalgDialect"]; + + let description = [{ + Fold transpose ops that feed `linalg.elementwise` into the elementwise op + by updating its indexing maps. `linalg.transpose` producers whose consumer + indexing map is the identity are absorbed, turning the permutation into + the elementwise map itself. Other operands remain untouched. + }]; } def LinalgDetensorizePass : InterfacePass<"linalg-detensorize", "FunctionOpInterface"> {