[Mlir-commits] [mlir] 2884421 - [mlir] Add additional traits to EqualOp and NotEqualOp.

[Adrian Kuegel]

Author: Adrian Kuegel
Date: 2021-05-21T15:04:26+02:00
New Revision: 28844212fe4054784549a1f552b78fe675fe4ebf

URL: https://github.com/llvm/llvm-project/commit/28844212fe4054784549a1f552b78fe675fe4ebf
DIFF: https://github.com/llvm/llvm-project/commit/28844212fe4054784549a1f552b78fe675fe4ebf.diff

LOG: [mlir] Add additional traits to EqualOp and NotEqualOp.

This matches the traits of the other binary ops.

Differential Revision: https://reviews.llvm.org/D102916

Added: 
    

Modified: 
    mlir/include/mlir/Dialect/Complex/IR/ComplexOps.td

Removed: 
    


################################################################################
diff  --git a/mlir/include/mlir/Dialect/Complex/IR/ComplexOps.td b/mlir/include/mlir/Dialect/Complex/IR/ComplexOps.td
index c31092a1a008..930d76884801 100644
--- a/mlir/include/mlir/Dialect/Complex/IR/ComplexOps.td
+++ b/mlir/include/mlir/Dialect/Complex/IR/ComplexOps.td
@@ -152,7 +152,9 @@ def ImOp : Complex_Op<"im",
 //===----------------------------------------------------------------------===//
 
 def EqualOp : Complex_Op<"eq",
-    [NoSideEffect, AllTypesMatch<["lhs", "rhs"]>]> {
+    [NoSideEffect, AllTypesMatch<["lhs", "rhs"]>,
+    DeclareOpInterfaceMethods<VectorUnrollOpInterface>] #
+    ElementwiseMappable.traits> {
   let summary = "computes whether two complex values are equal";
   let description = [{
     The `eq` op takes two complex numbers and returns whether they are equal.
@@ -194,7 +196,9 @@ def MulOp : ComplexArithmeticOp<"mul"> {
 //===----------------------------------------------------------------------===//
 
 def NotEqualOp : Complex_Op<"neq",
-    [NoSideEffect, AllTypesMatch<["lhs", "rhs"]>]> {
+    [NoSideEffect, AllTypesMatch<["lhs", "rhs"]>,
+    DeclareOpInterfaceMethods<VectorUnrollOpInterface>] #
+    ElementwiseMappable.traits> {
   let summary = "computes whether two complex values are not equal";
   let description = [{
     The `neq` op takes two complex numbers and returns whether they are not