This is an archive of the discontinued LLVM Phabricator instance.

Differential D145424

[InstCombine] Add tests for transforming `(icmp pred (and X, Y), X)`; NFC
ClosedPublic

Authored by goldstein.w.n on Mar 6 2023, 2:16 PM.

Details

Reviewers
nikic
spatel
majnemer
RKSimon
Commits
rG67ba29380809: [InstCombine] Add tests for transforming `(icmp pred (and X, Y), X)`; NFC

Diff Detail

Event Timeline

goldstein.w.n created this revision.Mar 6 2023, 2:16 PM
Herald added a project: Restricted Project. · View Herald TranscriptMar 6 2023, 2:16 PM
goldstein.w.n requested review of this revision.Mar 6 2023, 2:16 PM
Herald added a project: Restricted Project. · View Herald TranscriptMar 6 2023, 2:16 PM
Herald added a subscriber: llvm-commits. · View Herald Transcript
goldstein.w.n added a child revision: D145425: [InstCombine] Transform `(icmp ult/uge (and X, Y), X)` -> `(icmp ne/eq (and X, Y), X)`.Mar 6 2023, 2:18 PM
goldstein.w.n added reviewers: nikic, spatel, majnemer.
Herald added a subscriber: StephenFan. · View Herald TranscriptMar 6 2023, 2:19 PM
goldstein.w.n added a parent revision: D145281: [InstCombine] Add transforms for `(icmp spred (or X, Y), X)`.Mar 6 2023, 2:19 PM
Harbormaster completed remote builds in B217704: Diff 502802.Mar 6 2023, 7:11 PM
nikic added inline comments.May 28 2023, 3:43 AM
llvm/test/Transforms/InstCombine/icmp-of-and-x.ll
90

What are these barriers for?

goldstein.w.n updated this revision to Diff 526372.May 28 2023, 9:12 PM

Rebase

Harbormaster completed remote builds in B235102: Diff 526372.May 28 2023, 10:14 PM
goldstein.w.n added inline comments.Jun 29 2023, 1:35 AM
llvm/test/Transforms/InstCombine/icmp-of-and-x.ll
90

To prevent the branch from folding to a select.

goldstein.w.n added a parent revision: D145280: [InstCombine] Add transforms for `(icmp (xor X, Y), X)`.Aug 12 2023, 2:48 PM
goldstein.w.n removed a parent revision: D145281: [InstCombine] Add transforms for `(icmp spred (or X, Y), X)`.
goldstein.w.n mentioned this in D157312: [InstCombine] Add transforms for `(or/and (icmp eq/ne X,0),(icmp eq/ne X,Pow2OrZero))`.Aug 25 2023, 2:18 PM
goldstein.w.n removed a parent revision: D145280: [InstCombine] Add transforms for `(icmp (xor X, Y), X)`.Aug 28 2023, 9:27 PM
goldstein.w.n updated this revision to Diff 554160.Aug 28 2023, 9:28 PM

Rebase

Harbormaster completed remote builds in B255395: Diff 554160.Aug 28 2023, 10:35 PM
goldstein.w.n updated this revision to Diff 555218.Aug 31 2023, 6:28 PM

Rebase

RKSimon accepted this revision.Sep 1 2023, 5:50 AM
RKSimon added a subscriber: RKSimon.

LGTM

This revision is now accepted and ready to land.Sep 1 2023, 5:50 AM
goldstein.w.n updated this revision to Diff 556694.Sep 13 2023, 12:53 PM

Rebase

This revision was landed with ongoing or failed builds.Sep 13 2023, 1:50 PM
Closed by commit rG67ba29380809: [InstCombine] Add tests for transforming `(icmp pred (and X, Y), X)`; NFC (authored by goldstein.w.n). · Explain Why
This revision was automatically updated to reflect the committed changes.
goldstein.w.n added a commit: rG67ba29380809: [InstCombine] Add tests for transforming `(icmp pred (and X, Y), X)`; NFC.
Harbormaster completed remote builds in B257166: Diff 556694.Sep 13 2023, 2:26 PM

Revision Contents

PathSize
llvm/
test/
Transforms/
InstCombine/
300 lines

Diff 556715

llvm/test/Transforms/InstCombine/icmp-of-and-x.ll

  • This file was added.
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; RUN: opt < %s -passes=instcombine -S | FileCheck %s
declare i1 @barrier()
declare void @llvm.assume(i1)
define i1 @icmp_ult_x_y(i8 %x, i8 %y) {
; CHECK-LABEL: @icmp_ult_x_y(
; CHECK-NEXT: [[AND:%.*]] = and i8 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[Z:%.*]] = icmp ult i8 [[AND]], [[X]]
; CHECK-NEXT: ret i1 [[Z]]
;
%and = and i8 %x, %y
%z = icmp ult i8 %and, %x
ret i1 %z
}
define i1 @icmp_ult_x_y_2(i8 %xx, i8 %y) {
; CHECK-LABEL: @icmp_ult_x_y_2(
; CHECK-NEXT: [[X:%.*]] = mul i8 [[XX:%.*]], [[XX]]
; CHECK-NEXT: [[AND:%.*]] = and i8 [[X]], [[Y:%.*]]
; CHECK-NEXT: [[Z:%.*]] = icmp ugt i8 [[X]], [[AND]]
; CHECK-NEXT: ret i1 [[Z]]
;
%x = mul i8 %xx, %xx
%and = and i8 %x, %y
%z = icmp ugt i8 %x, %and
ret i1 %z
}
define <2 x i1> @icmp_uge_x_y(<2 x i8> %x, <2 x i8> %y) {
; CHECK-LABEL: @icmp_uge_x_y(
; CHECK-NEXT: [[AND:%.*]] = and <2 x i8> [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT: [[Z:%.*]] = icmp uge <2 x i8> [[AND]], [[X]]
; CHECK-NEXT: ret <2 x i1> [[Z]]
;
%and = and <2 x i8> %x, %y
%z = icmp uge <2 x i8> %and, %x
ret <2 x i1> %z
}
define i1 @icmp_uge_x_y_2(i8 %xx, i8 %y) {
; CHECK-LABEL: @icmp_uge_x_y_2(
; CHECK-NEXT: [[X:%.*]] = mul i8 [[XX:%.*]], [[XX]]
; CHECK-NEXT: [[AND:%.*]] = and i8 [[X]], [[Y:%.*]]
; CHECK-NEXT: [[Z:%.*]] = icmp ule i8 [[X]], [[AND]]
; CHECK-NEXT: ret i1 [[Z]]
;
%x = mul i8 %xx, %xx
%and = and i8 %x, %y
%z = icmp ule i8 %x, %and
ret i1 %z
}
define i1 @icmp_sge_x_negy(i8 %x, i8 %y) {
; CHECK-LABEL: @icmp_sge_x_negy(
; CHECK-NEXT: [[CY:%.*]] = icmp slt i8 [[Y:%.*]], 0
; CHECK-NEXT: call void @llvm.assume(i1 [[CY]])
; CHECK-NEXT: [[AND:%.*]] = and i8 [[X:%.*]], [[Y]]
; CHECK-NEXT: [[Z:%.*]] = icmp sge i8 [[AND]], [[X]]
; CHECK-NEXT: ret i1 [[Z]]
;
%cy = icmp slt i8 %y, 0
call void @llvm.assume(i1 %cy)
%and = and i8 %x, %y
%z = icmp sge i8 %and, %x
ret i1 %z
}
define i1 @icmp_slt_x_negy(i8 %x, i8 %y) {
; CHECK-LABEL: @icmp_slt_x_negy(
; CHECK-NEXT: [[CY:%.*]] = icmp slt i8 [[Y:%.*]], 0
; CHECK-NEXT: br i1 [[CY]], label [[NEGY:%.*]], label [[POSY:%.*]]
; CHECK: negy:
; CHECK-NEXT: [[AND:%.*]] = and i8 [[X:%.*]], [[Y]]
; CHECK-NEXT: [[Z:%.*]] = icmp slt i8 [[AND]], [[X]]
; CHECK-NEXT: ret i1 [[Z]]
; CHECK: posy:
; CHECK-NEXT: [[R:%.*]] = call i1 @barrier()
; CHECK-NEXT: ret i1 [[R]]
;
%cy = icmp slt i8 %y, 0
br i1 %cy, label %negy, label %posy
negy:
%and = and i8 %x, %y
%z = icmp slt i8 %and, %x
ret i1 %z
posy:
%r = call i1 @barrier()
ret i1 %r
nikicUnsubmitted
Not Done
ReplyInline Actions

What are these barriers for?

nikic: What are these barriers for?
goldstein.w.nAuthorUnsubmitted
Done
ReplyInline Actions

To prevent the branch from folding to a select.

goldstein.w.n: To prevent the branch from folding to a select.
}
define i1 @icmp_slt_x_negy_fail_maybe_zero(i8 %x, i8 %y) {
; CHECK-LABEL: @icmp_slt_x_negy_fail_maybe_zero(
; CHECK-NEXT: [[CY:%.*]] = icmp slt i8 [[Y:%.*]], 1
; CHECK-NEXT: br i1 [[CY]], label [[NEGY:%.*]], label [[POSY:%.*]]
; CHECK: negy:
; CHECK-NEXT: [[AND:%.*]] = and i8 [[X:%.*]], [[Y]]
; CHECK-NEXT: [[Z:%.*]] = icmp slt i8 [[AND]], [[X]]
; CHECK-NEXT: ret i1 [[Z]]
; CHECK: posy:
; CHECK-NEXT: [[R:%.*]] = call i1 @barrier()
; CHECK-NEXT: ret i1 [[R]]
;
%cy = icmp sle i8 %y, 0
br i1 %cy, label %negy, label %posy
negy:
%and = and i8 %x, %y
%z = icmp slt i8 %and, %x
ret i1 %z
posy:
%r = call i1 @barrier()
ret i1 %r
}
define i1 @icmp_sle_x_negy(i8 %x, i8 %yy) {
; CHECK-LABEL: @icmp_sle_x_negy(
; CHECK-NEXT: [[Y:%.*]] = or i8 [[YY:%.*]], -128
; CHECK-NEXT: [[AND:%.*]] = and i8 [[Y]], [[X:%.*]]
; CHECK-NEXT: [[Z:%.*]] = icmp sle i8 [[AND]], [[X]]
; CHECK-NEXT: ret i1 [[Z]]
;
%y = or i8 %yy, 128
%and = and i8 %y, %x
%z = icmp sle i8 %and, %x
ret i1 %z
}
define <2 x i1> @icmp_sgt_x_negy(<2 x i8> %x, <2 x i8> %yy) {
; CHECK-LABEL: @icmp_sgt_x_negy(
; CHECK-NEXT: [[Y:%.*]] = or <2 x i8> [[YY:%.*]], <i8 -128, i8 -128>
; CHECK-NEXT: [[AND:%.*]] = and <2 x i8> [[Y]], [[X:%.*]]
; CHECK-NEXT: [[Z:%.*]] = icmp sgt <2 x i8> [[AND]], [[X]]
; CHECK-NEXT: ret <2 x i1> [[Z]]
;
%y = or <2 x i8> %yy, <i8 128, i8 128>
%and = and <2 x i8> %y, %x
%z = icmp sgt <2 x i8> %and, %x
ret <2 x i1> %z
}
define <2 x i1> @icmp_sgt_x_negy_fail_partial(<2 x i8> %x, <2 x i8> %yy) {
; CHECK-LABEL: @icmp_sgt_x_negy_fail_partial(
; CHECK-NEXT: [[Y:%.*]] = or <2 x i8> [[YY:%.*]], <i8 -128, i8 4>
; CHECK-NEXT: [[AND:%.*]] = and <2 x i8> [[Y]], [[X:%.*]]
; CHECK-NEXT: [[Z:%.*]] = icmp sgt <2 x i8> [[AND]], [[X]]
; CHECK-NEXT: ret <2 x i1> [[Z]]
;
%y = or <2 x i8> %yy, <i8 128, i8 4>
%and = and <2 x i8> %y, %x
%z = icmp sgt <2 x i8> %and, %x
ret <2 x i1> %z
}
define <2 x i1> @icmp_sle_x_posy(<2 x i8> %x, <2 x i8> %yy) {
; CHECK-LABEL: @icmp_sle_x_posy(
; CHECK-NEXT: [[Y:%.*]] = and <2 x i8> [[YY:%.*]], <i8 127, i8 127>
; CHECK-NEXT: [[AND:%.*]] = and <2 x i8> [[Y]], [[X:%.*]]
; CHECK-NEXT: [[Z:%.*]] = icmp sle <2 x i8> [[AND]], [[X]]
; CHECK-NEXT: ret <2 x i1> [[Z]]
;
%y = and <2 x i8> %yy, <i8 127, i8 127>
%and = and <2 x i8> %y, %x
%z = icmp sle <2 x i8> %and, %x
ret <2 x i1> %z
}
define <2 x i1> @icmp_sle_x_posy_fail_partial(<2 x i8> %x, <2 x i8> %yy) {
; CHECK-LABEL: @icmp_sle_x_posy_fail_partial(
; CHECK-NEXT: [[Y:%.*]] = and <2 x i8> [[YY:%.*]], <i8 127, i8 -65>
; CHECK-NEXT: [[AND:%.*]] = and <2 x i8> [[Y]], [[X:%.*]]
; CHECK-NEXT: [[Z:%.*]] = icmp sle <2 x i8> [[AND]], [[X]]
; CHECK-NEXT: ret <2 x i1> [[Z]]
;
%y = and <2 x i8> %yy, <i8 127, i8 191>
%and = and <2 x i8> %y, %x
%z = icmp sle <2 x i8> %and, %x
ret <2 x i1> %z
}
define i1 @icmp_sgt_x_posy(i8 %x, i8 %y) {
; CHECK-LABEL: @icmp_sgt_x_posy(
; CHECK-NEXT: [[CY:%.*]] = icmp sgt i8 [[Y:%.*]], -1
; CHECK-NEXT: call void @llvm.assume(i1 [[CY]])
; CHECK-NEXT: [[AND:%.*]] = and i8 [[X:%.*]], [[Y]]
; CHECK-NEXT: [[Z:%.*]] = icmp sgt i8 [[AND]], [[X]]
; CHECK-NEXT: ret i1 [[Z]]
;
%cy = icmp sge i8 %y, 0
call void @llvm.assume(i1 %cy)
%and = and i8 %x, %y
%z = icmp sgt i8 %and, %x
ret i1 %z
}
define <2 x i1> @icmp_sgt_negx_y(<2 x i8> %xx, <2 x i8> %y) {
; CHECK-LABEL: @icmp_sgt_negx_y(
; CHECK-NEXT: [[X:%.*]] = or <2 x i8> [[XX:%.*]], <i8 -128, i8 -128>
; CHECK-NEXT: [[AND:%.*]] = and <2 x i8> [[X]], [[Y:%.*]]
; CHECK-NEXT: [[Z:%.*]] = icmp sgt <2 x i8> [[AND]], [[X]]
; CHECK-NEXT: ret <2 x i1> [[Z]]
;
%x = or <2 x i8> %xx, <i8 128, i8 128>
%and = and <2 x i8> %x, %y
%z = icmp sgt <2 x i8> %and, %x
ret <2 x i1> %z
}
define i1 @icmp_sle_negx_y(i8 %x, i8 %y) {
; CHECK-LABEL: @icmp_sle_negx_y(
; CHECK-NEXT: [[CX:%.*]] = icmp slt i8 [[X:%.*]], 0
; CHECK-NEXT: call void @llvm.assume(i1 [[CX]])
; CHECK-NEXT: [[AND:%.*]] = and i8 [[X]], [[Y:%.*]]
; CHECK-NEXT: [[Z:%.*]] = icmp sle i8 [[AND]], [[X]]
; CHECK-NEXT: ret i1 [[Z]]
;
%cx = icmp slt i8 %x, 0
call void @llvm.assume(i1 %cx)
%and = and i8 %x, %y
%z = icmp sle i8 %and, %x
ret i1 %z
}
define i1 @icmp_sle_negx_y_fail_maybe_zero(i8 %x, i8 %y) {
; CHECK-LABEL: @icmp_sle_negx_y_fail_maybe_zero(
; CHECK-NEXT: [[CX:%.*]] = icmp slt i8 [[X:%.*]], 1
; CHECK-NEXT: call void @llvm.assume(i1 [[CX]])
; CHECK-NEXT: [[AND:%.*]] = and i8 [[X]], [[Y:%.*]]
; CHECK-NEXT: [[Z:%.*]] = icmp sle i8 [[AND]], [[X]]
; CHECK-NEXT: ret i1 [[Z]]
;
%cx = icmp sle i8 %x, 0
call void @llvm.assume(i1 %cx)
%and = and i8 %x, %y
%z = icmp sle i8 %and, %x
ret i1 %z
}
define i1 @icmp_eq_x_invertable_y_todo(i8 %x, i1 %y) {
; CHECK-LABEL: @icmp_eq_x_invertable_y_todo(
; CHECK-NEXT: [[YY:%.*]] = select i1 [[Y:%.*]], i8 7, i8 24
; CHECK-NEXT: [[AND:%.*]] = and i8 [[YY]], [[X:%.*]]
; CHECK-NEXT: [[R:%.*]] = icmp eq i8 [[AND]], [[X]]
; CHECK-NEXT: ret i1 [[R]]
;
%yy = select i1 %y, i8 7, i8 24
%and = and i8 %x, %yy
%r = icmp eq i8 %x, %and
ret i1 %r
}
define i1 @icmp_eq_x_invertable_y(i8 %x, i8 %y) {
; CHECK-LABEL: @icmp_eq_x_invertable_y(
; CHECK-NEXT: [[YY:%.*]] = xor i8 [[Y:%.*]], -1
; CHECK-NEXT: [[AND:%.*]] = and i8 [[YY]], [[X:%.*]]
; CHECK-NEXT: [[R:%.*]] = icmp eq i8 [[AND]], [[X]]
; CHECK-NEXT: ret i1 [[R]]
;
%yy = xor i8 %y, -1
%and = and i8 %x, %yy
%r = icmp eq i8 %x, %and
ret i1 %r
}
define i1 @icmp_eq_x_invertable_y2_todo(i8 %x, i1 %y) {
; CHECK-LABEL: @icmp_eq_x_invertable_y2_todo(
; CHECK-NEXT: [[YY:%.*]] = select i1 [[Y:%.*]], i8 7, i8 24
; CHECK-NEXT: [[AND:%.*]] = and i8 [[YY]], [[X:%.*]]
; CHECK-NEXT: [[R:%.*]] = icmp eq i8 [[YY]], [[AND]]
; CHECK-NEXT: ret i1 [[R]]
;
%yy = select i1 %y, i8 7, i8 24
%and = and i8 %x, %yy
%r = icmp eq i8 %yy, %and
ret i1 %r
}
define i1 @icmp_eq_x_invertable_y2(i8 %x, i8 %y) {
; CHECK-LABEL: @icmp_eq_x_invertable_y2(
; CHECK-NEXT: [[YY:%.*]] = xor i8 [[Y:%.*]], -1
; CHECK-NEXT: [[AND:%.*]] = and i8 [[YY]], [[X:%.*]]
; CHECK-NEXT: [[R:%.*]] = icmp eq i8 [[AND]], [[YY]]
; CHECK-NEXT: ret i1 [[R]]
;
%yy = xor i8 %y, -1
%and = and i8 %x, %yy
%r = icmp eq i8 %yy, %and
ret i1 %r
}
define i1 @icmp_eq_x_invertable_y_fail_immconstant(i8 %x, i8 %y) {
; CHECK-LABEL: @icmp_eq_x_invertable_y_fail_immconstant(
; CHECK-NEXT: [[AND:%.*]] = and i8 [[X:%.*]], 7
; CHECK-NEXT: [[R:%.*]] = icmp eq i8 [[AND]], 7
; CHECK-NEXT: ret i1 [[R]]
;
%and = and i8 %x, 7
%r = icmp eq i8 %and, 7
ret i1 %r
}