[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