[Mlir-commits] [mlir] [mlir][sparse] fuse concat and extract_slice op if possible. (PR #89825)

Aart Bik llvmlistbot at llvm.org
Wed Apr 24 11:04:59 PDT 2024


================
@@ -209,6 +209,86 @@ static void concatSizesFromInputs(OpBuilder &builder,
 
 namespace {
 
+/// TODO: move it to tensor dialect instead.
+///
+/// Fold `tensor.concat` and `tensor.extract_slice`
+///
+/// %concat = tensor.concat dim(2) %t0, %t1
----------------
aartbik wrote:

+1 on that, just a CHECK test in  test/Dialect/SparseTensor/ with a simple before after pattern


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


More information about the Mlir-commits mailing list