[Mlir-commits] [mlir] [mlir][Tensor] Fold destination-style ops into `tensor.unpack` operation. (PR #71468)

lorenzo chelini llvmlistbot at llvm.org
Mon Nov 6 23:39:48 PST 2023


================
@@ -1861,3 +1861,19 @@ func.func @invalid_empty_negative_size() -> (tensor<4x5x?xf32>) {
   %1 = tensor.empty(%0) : tensor<4x5x?xf32>
   return %1 : tensor<4x5x?xf32>
 }
+
+// -----
+
+// Fold DstStyleOp -> tensor.unpack operations.
+func.func @fold_dst_style_ops_into_unpack(%arg0 : tensor<?x?x16x64xf32>, %init : tensor<?x?xf32>) -> tensor<?x?xf32> {
----------------
chelini wrote:

nit: I would use `arg1` instead of `init` or update the checks line.

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


More information about the Mlir-commits mailing list