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

Nicolas Vasilache llvmlistbot at llvm.org
Mon Nov 20 04:50:32 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);
----------------
nicolasvasilache wrote:

can we split this in a followup ?

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


More information about the Mlir-commits mailing list