[mlir] Update custom<DynamicIndexList> for Pack/Unpack.
authorAlexander Belyaev <pifon@google.com>
Tue, 22 Nov 2022 09:58:23 +0000 (10:58 +0100)
committerAlexander Belyaev <pifon@google.com>
Tue, 22 Nov 2022 09:58:23 +0000 (10:58 +0100)
mlir/include/mlir/Dialect/Tensor/IR/TensorOps.td

index 24acd4b..7af19a7 100644 (file)
@@ -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<DynamicIndexList>($inner_tiles, $static_inner_tiles,
-                             "ShapedType::kDynamic")
+    custom<DynamicIndexList>($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<DynamicIndexList>($inner_tiles, $static_inner_tiles,
-                             "ShapedType::kDynamic")
+    custom<DynamicIndexList>($inner_tiles, $static_inner_tiles)
     `into` $dest attr-dict `:` type($source) `->` type($dest)
   }];