[Mlir-commits] [mlir] [MLIR][Linalg] Fix insert_slice fusion with rank reduction (PR #130961)

[Andrzej Warzyński]

================
@@ -43,6 +43,11 @@ FailureOr<RankedTensorType>
 computeTransposedType(RankedTensorType rankedTensorType,
                       ArrayRef<int64_t> transposeVector);
 
+/// Create tensor.collapse_shape to drop unit dimensions in `dropDims` in tensor
+/// `from`.
+CollapseShapeOp dropGivenUnitDims(OpBuilder &b, Location loc, Value from,
+                                  const llvm::SmallBitVector &dropDims);
----------------
banach-space wrote:

[nit] `from` -> `src` to match the Op definition: https://github.com/llvm/llvm-project/blob/dc0dcab397ae3de38141e1995e4b4e5e3bb98660/mlir/include/mlir/Dialect/Tensor/IR/TensorOps.td#L1186

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