intel/llvm
1
0
Fork 0
mirror of https://github.com/intel/llvm.git synced 2026-01-24 00:20:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

[InstCombine] Preserve nneg in foldLogicCastConstant (#157865)

Browse Source 
This patch makes use of the new public helper function to preserve nneg
in `foldLogicCastConstant`.
This commit is contained in:
Hongyu Chen
2025-09-11 00:55:02 +08:00
committed by GitHub
parent b574e63f9f
commit c511e1a0f8
2 changed files with 72 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
llvm/lib/Transforms/InstCombine/InstCombineAndOrXor.cpp
View File

@@ -1801,10 +1801,14 @@ static Instruction *foldLogicCastConstant(BinaryOperator &Logic, CastInst *Cast,
Value *X;
auto &DL = IC.getDataLayout();
if (match(Cast, m_OneUse(m_ZExt(m_Value(X))))) {
if (Constant *TruncC = getLosslessUnsignedTrunc(C, SrcTy, DL)) {
PreservedCastFlags Flags;
if (Constant *TruncC = getLosslessUnsignedTrunc(C, SrcTy, DL, &Flags)) {
// LogicOpc (zext X), C --> zext (LogicOpc X, C)
Value *NewOp = IC.Builder.CreateBinOp(LogicOpc, X, TruncC);
return new ZExtInst(NewOp, DestTy);
auto *ZExt = new ZExtInst(NewOp, DestTy);
ZExt->setNonNeg(Flags.NNeg);
ZExt->andIRFlags(Cast);
return ZExt;
}
}

66
llvm/test/Transforms/InstCombine/or.ll
View File

@@ -2047,3 +2047,69 @@ define i1 @or_truncs(i8 %x) {
%or1 = or i1 %trunc1, %trunc2
ret i1 %or1
}
define i32 @or_zext_constant(i8 %a) {
; CHECK-LABEL: @or_zext_constant(
; CHECK-NEXT: [[TMP1:%.*]] = or i8 [[A:%.*]], 1
; CHECK-NEXT: [[AND:%.*]] = zext i8 [[TMP1]] to i32
; CHECK-NEXT: ret i32 [[AND]]
;
%zext = zext i8 %a to i32
%or = or i32 %zext, 1
ret i32 %or
}
define i32 @or_zext_minus_constant(i8 %a) {
; CHECK-LABEL: @or_zext_minus_constant(
; CHECK-NEXT: [[OR:%.*]] = zext i8 [[TMP1:%.*]] to i32
; CHECK-NEXT: [[OR1:%.*]] = or i32 [[OR]], -9
; CHECK-NEXT: ret i32 [[OR1]]
;
%zext = zext i8 %a to i32
%or = or i32 %zext, -9
ret i32 %or
}
define i32 @or_zext_nneg_constant(i8 %a) {
; CHECK-LABEL: @or_zext_nneg_constant(
; CHECK-NEXT: [[TMP1:%.*]] = or i8 [[A:%.*]], 9
; CHECK-NEXT: [[AND:%.*]] = zext nneg i8 [[TMP1]] to i32
; CHECK-NEXT: ret i32 [[AND]]
;
%zext = zext nneg i8 %a to i32
%or = or i32 %zext, 9
ret i32 %or
}
define <4 x i32> @or_zext_nneg_constant_splat(<4 x i8> %a) {
; CHECK-LABEL: @or_zext_nneg_constant_splat(
; CHECK-NEXT: [[TMP1:%.*]] = or <4 x i8> [[A:%.*]], splat (i8 9)
; CHECK-NEXT: [[OR:%.*]] = zext nneg <4 x i8> [[TMP1]] to <4 x i32>
; CHECK-NEXT: ret <4 x i32> [[OR]]
;
%zext = zext nneg <4 x i8> %a to <4 x i32>
%or = or <4 x i32> %zext, splat (i32 9)
ret <4 x i32> %or
}
define i32 @or_zext_nneg_minus_constant(i8 %a) {
; CHECK-LABEL: @or_zext_nneg_minus_constant(
; CHECK-NEXT: [[TMP1:%.*]] = or i8 [[A:%.*]], -9
; CHECK-NEXT: [[ZEXT:%.*]] = sext i8 [[TMP1]] to i32
; CHECK-NEXT: ret i32 [[ZEXT]]
;
%zext = zext nneg i8 %a to i32
%or = or i32 %zext, -9
ret i32 %or
}
define <4 x i32> @or_zext_nneg_minus_constant_splat(<4 x i8> %a) {
; CHECK-LABEL: @or_zext_nneg_minus_constant_splat(
; CHECK-NEXT: [[TMP1:%.*]] = or <4 x i8> [[A:%.*]], splat (i8 -9)
; CHECK-NEXT: [[OR:%.*]] = sext <4 x i8> [[TMP1]] to <4 x i32>
; CHECK-NEXT: ret <4 x i32> [[OR]]
;
%zext = zext nneg <4 x i8> %a to <4 x i32>
%or = or <4 x i32> %zext, splat (i32 -9)
ret <4 x i32> %or
}