[Mlir-commits] [mlir] Update VectorOps.td (PR #65533)

[Frank Wessels]

https://github.com/fwessels created https://github.com/llvm/llvm-project/pull/65533:

Add MaskableOpInterface for scatter operation (as well as VectorUnrollOpInterface).

>From 10afe4dd7e3ba11becbecce9bc9fb4973a867949 Mon Sep 17 00:00:00 2001
From: Frank Wessels <fwessels at xs4all.nl>
Date: Wed, 6 Sep 2023 14:19:44 -0700
Subject: [PATCH] Update VectorOps.td

Add MaskableOpInterface for scatter operation
---
 mlir/include/mlir/Dialect/Vector/IR/VectorOps.td | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/mlir/include/mlir/Dialect/Vector/IR/VectorOps.td b/mlir/include/mlir/Dialect/Vector/IR/VectorOps.td
index bf42b4053ac05b8..34386ebf4688eea 100644
--- a/mlir/include/mlir/Dialect/Vector/IR/VectorOps.td
+++ b/mlir/include/mlir/Dialect/Vector/IR/VectorOps.td
@@ -1901,7 +1901,10 @@ def Vector_GatherOp :
 }
 
 def Vector_ScatterOp :
-  Vector_Op<"scatter">,
+  Vector_Op<"scatter">, [
+    DeclareOpInterfaceMethods<MaskableOpInterface>,
+    DeclareOpInterfaceMethods<VectorUnrollOpInterface, ["getShapeForUnroll"]>
+  ]>,
     Arguments<(ins Arg<AnyMemRef, "", [MemWrite]>:$base,
                Variadic<Index>:$indices,
                VectorOfRankAndType<[1], [AnyInteger, Index]>:$index_vec,