This is an archive of the discontinued LLVM Phabricator instance.

	Show First 20 Lines • Show All 109 Lines • ▼ Show 20 Lines
	// not(cmpi(pred, a, b)) -> cmpi(~pred, a, b), where not(x) is xori(x, 1)			// not(cmpi(pred, a, b)) -> cmpi(~pred, a, b), where not(x) is xori(x, 1)
	def InvertPredicate : NativeCodeCall<"invertPredicate($0)">;			def InvertPredicate : NativeCodeCall<"invertPredicate($0)">;
	def XOrINotCmpI :			def XOrINotCmpI :
	Pat<(Arith_XOrIOp			Pat<(Arith_XOrIOp
	(Arith_CmpIOp $pred, $a, $b),			(Arith_CmpIOp $pred, $a, $b),
	(Arith_ConstantOp ConstantAttr<I1Attr, "1">)),			(Arith_ConstantOp ConstantAttr<I1Attr, "1">)),
	(Arith_CmpIOp (InvertPredicate $pred), $a, $b)>;			(Arith_CmpIOp (InvertPredicate $pred), $a, $b)>;

				// xor extui(x), extui(y) -> extui(xor(x,y))
				def XOrIOfExtUI :
				Pat<(Arith_XOrIOp (Arith_ExtUIOp $x), (Arith_ExtUIOp $y)), (Arith_ExtUIOp (Arith_XOrIOp $x, $y)),
				[(Constraint<CPred<"$0.getType() == $1.getType()">> $x, $y)]>;

				// xor extsi(x), extsi(y) -> extsi(xor(x,y))
				def XOrIOfExtSI :
				Pat<(Arith_XOrIOp (Arith_ExtSIOp $x), (Arith_ExtSIOp $y)), (Arith_ExtSIOp (Arith_XOrIOp $x, $y)),
				[(Constraint<CPred<"$0.getType() == $1.getType()">> $x, $y)]>;

	//===----------------------------------------------------------------------===//			//===----------------------------------------------------------------------===//
	// CmpIOp			// CmpIOp
	//===----------------------------------------------------------------------===//			//===----------------------------------------------------------------------===//

	// cmpi(== or !=, a ext iNN, b ext iNN) == cmpi(== or !=, a, b)			// cmpi(== or !=, a ext iNN, b ext iNN) == cmpi(== or !=, a, b)
	def CmpIExtSI :			def CmpIExtSI :
	Pat<(Arith_CmpIOp $pred,			Pat<(Arith_CmpIOp $pred,
	(Arith_ExtSIOp $a),			(Arith_ExtSIOp $a),
	▲ Show 20 Lines • Show All 110 Lines • Show Last 20 Lines

mlir/lib/Dialect/Arith/IR/ArithOps.cpp

Show First 20 Lines • Show All 701 Lines • ▼ Show 20 Lines	OpFoldResult arith::XOrIOp::fold(ArrayRef<Attribute> operands) {
}		}

return constFoldBinaryOp<IntegerAttr>(		return constFoldBinaryOp<IntegerAttr>(
operands, [](APInt a, const APInt &b) { return std::move(a) ^ b; });		operands, [](APInt a, const APInt &b) { return std::move(a) ^ b; });
}		}

void arith::XOrIOp::getCanonicalizationPatterns(RewritePatternSet &patterns,		void arith::XOrIOp::getCanonicalizationPatterns(RewritePatternSet &patterns,
MLIRContext *context) {		MLIRContext *context) {
patterns.add<XOrINotCmpI>(context);		patterns.add<XOrINotCmpI, XOrIOfExtUI, XOrIOfExtSI>(context);
}		}

//===----------------------------------------------------------------------===//		//===----------------------------------------------------------------------===//
// NegFOp		// NegFOp
//===----------------------------------------------------------------------===//		//===----------------------------------------------------------------------===//

OpFoldResult arith::NegFOp::fold(ArrayRef<Attribute> operands) {		OpFoldResult arith::NegFOp::fold(ArrayRef<Attribute> operands) {
/// negf(negf(x)) -> x		/// negf(negf(x)) -> x
▲ Show 20 Lines • Show All 1,544 Lines • Show Last 20 Lines

mlir/test/Dialect/Arith/canonicalize.mlir

	Show First 20 Lines • Show All 311 Lines • ▼ Show 20 Lines
	// CHECK: return %[[ext]]			// CHECK: return %[[ext]]
	func.func @orOfExtUI(%arg0: i8, %arg1: i8) -> i64 {			func.func @orOfExtUI(%arg0: i8, %arg1: i8) -> i64 {
	%ext0 = arith.extui %arg0 : i8 to i64			%ext0 = arith.extui %arg0 : i8 to i64
	%ext1 = arith.extui %arg1 : i8 to i64			%ext1 = arith.extui %arg1 : i8 to i64
	%res = arith.ori %ext0, %ext1 : i64			%res = arith.ori %ext0, %ext1 : i64
	return %res : i64			return %res : i64
	}			}

				// CHECK-LABEL: @xorOfExtSI
				// CHECK: %[[comb:.+]] = arith.xori %arg0, %arg1 : i8
				// CHECK: %[[ext:.+]] = arith.extsi %[[comb]] : i8 to i64
				// CHECK: return %[[ext]]
				func.func @xorOfExtSI(%arg0: i8, %arg1: i8) -> i64 {
				%ext0 = arith.extsi %arg0 : i8 to i64
				%ext1 = arith.extsi %arg1 : i8 to i64
				%res = arith.xori %ext0, %ext1 : i64
				return %res : i64
				}

				// CHECK-LABEL: @xorOfExtUI
				// CHECK: %[[comb:.+]] = arith.xori %arg0, %arg1 : i8
				// CHECK: %[[ext:.+]] = arith.extui %[[comb]] : i8 to i64
				// CHECK: return %[[ext]]
				func.func @xorOfExtUI(%arg0: i8, %arg1: i8) -> i64 {
				%ext0 = arith.extui %arg0 : i8 to i64
				%ext1 = arith.extui %arg1 : i8 to i64
				%res = arith.xori %ext0, %ext1 : i64
				return %res : i64
				}

	// -----			// -----

	// CHECK-LABEL: @indexCastOfSignExtend			// CHECK-LABEL: @indexCastOfSignExtend
	// CHECK: %[[res:.+]] = arith.index_cast %arg0 : i8 to index			// CHECK: %[[res:.+]] = arith.index_cast %arg0 : i8 to index
	// CHECK: return %[[res]]			// CHECK: return %[[res]]
	func.func @indexCastOfSignExtend(%arg0: i8) -> index {			func.func @indexCastOfSignExtend(%arg0: i8) -> index {
	%ext = arith.extsi %arg0 : i8 to i16			%ext = arith.extsi %arg0 : i8 to i16
	%idx = arith.index_cast %ext : i16 to index			%idx = arith.index_cast %ext : i16 to index
	▲ Show 20 Lines • Show All 1,399 Lines • Show Last 20 Lines

This is an archive of the discontinued LLVM Phabricator instance.

[MLIR][Arith] Canonicalize xor with extClosedPublic

Details

Diff Detail

Event Timeline

Revision Contents

Diff 480048

mlir/lib/Dialect/Arith/IR/ArithCanonicalization.td

mlir/lib/Dialect/Arith/IR/ArithOps.cpp

mlir/test/Dialect/Arith/canonicalize.mlir

[MLIR][Arith] Canonicalize xor with ext
ClosedPublic