[Mlir-commits] [mlir] 38ff7e1 - [mlir][tosa] Add several binary elementwise to the list of broadcastable ops.

[Rob Suderman]

Author: natashaknk
Date: 2021-09-20T16:07:35-07:00
New Revision: 38ff7e11c04e760570e3cb517f8b78d554c65386

URL: https://github.com/llvm/llvm-project/commit/38ff7e11c04e760570e3cb517f8b78d554c65386
DIFF: https://github.com/llvm/llvm-project/commit/38ff7e11c04e760570e3cb517f8b78d554c65386.diff

LOG: [mlir][tosa] Add several binary elementwise to the list of broadcastable ops.

Reviewed By: rsuderman

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

Added: 
    

Modified: 
    mlir/lib/Dialect/Tosa/Transforms/TosaMakeBroadcastable.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/Tosa/Transforms/TosaMakeBroadcastable.cpp b/mlir/lib/Dialect/Tosa/Transforms/TosaMakeBroadcastable.cpp
index 4e3706ec992b9..14bd29f4c1a9c 100644
--- a/mlir/lib/Dialect/Tosa/Transforms/TosaMakeBroadcastable.cpp
+++ b/mlir/lib/Dialect/Tosa/Transforms/TosaMakeBroadcastable.cpp
@@ -269,6 +269,9 @@ struct TosaMakeBroadcastable
     RewritePatternSet patterns(func.getContext());
     MLIRContext *ctx = func.getContext();
     // Add the generated patterns to the list.
+    patterns.add<ConvertTosaOp<tosa::BitwiseAndOp>>(ctx);
+    patterns.add<ConvertTosaOp<tosa::BitwiseOrOp>>(ctx);
+    patterns.add<ConvertTosaOp<tosa::BitwiseXorOp>>(ctx);
     patterns.add<ConvertTosaOp<tosa::AddOp>>(ctx);
     patterns.add<ConvertTosaOp<tosa::SubOp>>(ctx);
     patterns.add<ConvertTosaOp<tosa::MulOp>>(ctx);
@@ -281,6 +284,10 @@ struct TosaMakeBroadcastable
     patterns.add<ConvertTosaOp<tosa::LogicalLeftShiftOp>>(ctx);
     patterns.add<ConvertTosaOp<tosa::ArithmeticRightShiftOp>>(ctx);
     patterns.add<ConvertTosaOp<tosa::LogicalRightShiftOp>>(ctx);
+    patterns.add<ConvertTosaOp<tosa::LogicalAndOp>>(ctx);
+    patterns.add<ConvertTosaOp<tosa::LogicalOrOp>>(ctx);
+    patterns.add<ConvertTosaOp<tosa::LogicalXorOp>>(ctx);
+    patterns.add<ConvertTosaOp<tosa::PowOp>>(ctx);
     (void)applyPatternsAndFoldGreedily(func, std::move(patterns));
   }
 };