[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