66e0f66d8f
Eliminate the subsequent applications of `size_to_index` and `index_to_size`. Differential Revision: https://reviews.llvm.org/D82083
29 lines
801 B
TableGen
29 lines
801 B
TableGen
include "mlir/Dialect/Shape/IR/ShapeOps.td"
|
|
|
|
def EqualBinaryOperands : Constraint<CPred<"$0 == $1">>;
|
|
|
|
def AllInputShapesEq : Constraint<CPred< [{
|
|
llvm::all_of($0, [&](mlir::Value val) {
|
|
return $0[0] == val;
|
|
})
|
|
}]>>;
|
|
|
|
// Canonicalization patterns.
|
|
|
|
def CstrBroadcastableEqOps : Pat<(Shape_CstrBroadcastableOp:$op $lhs, $rhs),
|
|
(Shape_ConstWitnessOp ConstBoolAttrTrue),
|
|
[(EqualBinaryOperands $lhs, $rhs)]>;
|
|
|
|
def CstrEqEqOps : Pat<(Shape_CstrEqOp:$op $shapes),
|
|
(Shape_ConstWitnessOp ConstBoolAttrTrue),
|
|
[(AllInputShapesEq $shapes)]>;
|
|
|
|
def IndexToSizeToIndexCanonicalization : Pat<
|
|
(Shape_SizeToIndexOp (Shape_IndexToSizeOp $arg)),
|
|
(replaceWithValue $arg)>;
|
|
|
|
def SizeToIndexToSizeCanonicalization : Pat<
|
|
(Shape_IndexToSizeOp (Shape_SizeToIndexOp $arg)),
|
|
(replaceWithValue $arg)>;
|
|
|