[mlir][Linalg] NFC: Refactor fusion of LinalgOp with TensorReshapeOp …

…by expansion. Change the implementation of LinalgOp with TensorReshapeOp by expansion to be more modular and easier to follow. Differential Revision: https://reviews.llvm.org/D93748
llvm · Jan 8, 2021 · fa8c397 · fa8c397
1 parent a8bda3d
commit fa8c397
Show file tree

Hide file tree

Showing 2 changed files with 249 additions and 162 deletions.
diff --git a/mlir/include/mlir/Dialect/Linalg/IR/LinalgOps.h b/mlir/include/mlir/Dialect/Linalg/IR/LinalgOps.h
@@ -56,6 +56,7 @@ SmallVector<Value, 4> applyMapToValues(OpBuilder &b, Location loc,
 LoopRangeBuilder defaultLoopRangesBuilder(LinalgOp op);
 
 using ReassociationIndices = SmallVector<int64_t, 2>;
+using ReassociationIndicesRef = ArrayRef<int64_t>;
 using ReassociationExprs = SmallVector<AffineExpr, 2>;
 
 /// Returns the name mangled library call name to disambiguate between different