From ae00ae487c9a24651039c50b44ef9176b8c9d499 Mon Sep 17 00:00:00 2001 From: Roman Dzhabarov Date: Tue, 20 Oct 2020 10:46:19 -0700 Subject: [PATCH] [mlir] Simplify DDR matching patterns with equal operands for operators. This https://reviews.llvm.org/D89254 diff introduced implicit matching between same name arguments. Modify usages accordingly. PiperOrigin-RevId: 338090110 --- lib/Dialect/mhlo/IR/hlo_patterns.td | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/lib/Dialect/mhlo/IR/hlo_patterns.td b/lib/Dialect/mhlo/IR/hlo_patterns.td index b8b6cb8..bdb3e3c 100644 --- a/lib/Dialect/mhlo/IR/hlo_patterns.td +++ b/lib/Dialect/mhlo/IR/hlo_patterns.td @@ -18,15 +18,13 @@ limitations under the License. include "mlir/Dialect/Shape/IR/ShapeOps.td" include "mlir-hlo/Dialect/mhlo/IR/hlo_ops.td" -def EqualBinaryOperands : Constraint>; - // Canonicalization patterns. def DynamicBroadcastToOwnShape_1 : Pat< - (HLO_DynamicBroadcastInDimOp:$op $arg0, - (Shape_ToExtentTensorOp (Shape_ShapeOfOp $arg1)), $attr), - (replaceWithValue $arg0), [(EqualBinaryOperands $arg0, $arg1)]>; + (HLO_DynamicBroadcastInDimOp:$op $x, + (Shape_ToExtentTensorOp (Shape_ShapeOfOp $x)), $attr), + (replaceWithValue $x)>; def DynamicBroadcastToOwnShape_2 : Pat< - (HLO_DynamicBroadcastInDimOp:$op $arg0, (Shape_ShapeOfOp $arg1), $attr), - (replaceWithValue $arg0), [(EqualBinaryOperands $arg0, $arg1)]>; + (HLO_DynamicBroadcastInDimOp:$op $x, (Shape_ShapeOfOp $x), $attr), + (replaceWithValue $x)>;