From a3c3e59ca5be12a39882aab87ddc23d5146cbcbc Mon Sep 17 00:00:00 2001 From: Alexander Belyaev Date: Tue, 22 Nov 2022 10:58:23 +0100 Subject: [PATCH] [mlir] Update custom for Pack/Unpack. --- mlir/include/mlir/Dialect/Tensor/IR/TensorOps.td | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/mlir/include/mlir/Dialect/Tensor/IR/TensorOps.td b/mlir/include/mlir/Dialect/Tensor/IR/TensorOps.td index 24acd4b..7af19a7 100644 --- a/mlir/include/mlir/Dialect/Tensor/IR/TensorOps.td +++ b/mlir/include/mlir/Dialect/Tensor/IR/TensorOps.td @@ -1753,8 +1753,7 @@ def Tensor_PackOp : Tensor_RelayoutOp<"pack", [ (`outer_dims_perm` `=` $outer_dims_perm^)? `inner_dims_pos` `=` $inner_dims_pos `inner_tiles` `=` - custom($inner_tiles, $static_inner_tiles, - "ShapedType::kDynamic") + custom($inner_tiles, $static_inner_tiles) `into` $dest attr-dict `:` type($source) `->` type($dest) }]; @@ -1808,8 +1807,7 @@ def Tensor_UnPackOp : Tensor_RelayoutOp<"unpack"> { (`outer_dims_perm` `=` $outer_dims_perm^)? `inner_dims_pos` `=` $inner_dims_pos `inner_tiles` `=` - custom($inner_tiles, $static_inner_tiles, - "ShapedType::kDynamic") + custom($inner_tiles, $static_inner_tiles) `into` $dest attr-dict `:` type($source) `->` type($dest) }]; -- 2.7.4