[Mlir-commits] [mlir] [mlir][tensor] Add a tensor.concat operation (PR #72779)

Quinn Dawkins llvmlistbot at llvm.org
Mon Nov 20 05:56:14 PST 2023


================
@@ -67,6 +67,11 @@ void populateReassociativeReshapeFoldingPatterns(RewritePatternSet &patterns);
 void populateFoldTensorEmptyPatterns(RewritePatternSet &patterns,
                                      bool foldSingleUseOnly = false);
 
+/// Populates `patterns` with patterns that decompose `tensor.concat` into
+/// `tensor.empty` of a tensor of the concatenated size, followed by a chain
+/// of `tensor.insert_slice` operations on the inputs.
+void populateDecomposeTensorConcatPatterns(RewritePatternSet &patterns);
----------------
qedawkins wrote:

Sure. Should I split out the implementation of `ReifyRankedShapedTypeOpInterface` as well? I'm not sure how to test this interface in isolation.

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


More information about the Mlir-commits mailing list