From d841c72e09c81a5892a52e3c3200af838f6a2a69 Mon Sep 17 00:00:00 2001
From: Usman Nadeem <mnadeem@quicinc.com>
Date: Sat, 18 Sep 2021 11:28:49 -0700
Subject: [PATCH] Precommit tests for D109807 "[InstCombine] Narrow type of
 logical operation chains when possible"

Change-Id: Iae9bf18619e4926301a866c7e2bd38ced524804e
---
 llvm/test/Transforms/InstCombine/and-xor-or.ll | 132 +++++++++++++++++++++++++
 1 file changed, 132 insertions(+)

diff --git a/llvm/test/Transforms/InstCombine/and-xor-or.ll b/llvm/test/Transforms/InstCombine/and-xor-or.ll
index 4cb88d0..993d835 100644
--- a/llvm/test/Transforms/InstCombine/and-xor-or.ll
+++ b/llvm/test/Transforms/InstCombine/and-xor-or.ll
@@ -384,3 +384,135 @@ define i8 @xor_lshr_multiuse(i8 %x, i8 %y, i8 %z, i8 %shamt) {
   ret i8 %r2
 }
 
+; Reassociate chains of extend(X) | (extend(Y) | Z).
+; Check that logical op is performed on a smaller type and then extended.
+
+define i64 @sext_or_chain(i64 %a, i16 %b, i16 %c) {
+; CHECK-LABEL: @sext_or_chain(
+; CHECK-NEXT:    [[CONV:%.*]] = sext i16 [[B:%.*]] to i64
+; CHECK-NEXT:    [[CONV2:%.*]] = sext i16 [[C:%.*]] to i64
+; CHECK-NEXT:    [[OR:%.*]] = or i64 [[CONV]], [[A:%.*]]
+; CHECK-NEXT:    [[OR2:%.*]] = or i64 [[OR]], [[CONV2]]
+; CHECK-NEXT:    ret i64 [[OR2]]
+;
+  %conv = sext i16 %b to i64
+  %conv2 = sext i16 %c to i64
+  %or = or i64 %a, %conv
+  %or2 = or i64 %or, %conv2
+  ret i64 %or2
+}
+
+define i64 @zext_or_chain(i64 %a, i16 %b, i16 %c) {
+; CHECK-LABEL: @zext_or_chain(
+; CHECK-NEXT:    [[CONV:%.*]] = zext i16 [[B:%.*]] to i64
+; CHECK-NEXT:    [[CONV2:%.*]] = zext i16 [[C:%.*]] to i64
+; CHECK-NEXT:    [[OR:%.*]] = or i64 [[CONV]], [[A:%.*]]
+; CHECK-NEXT:    [[OR2:%.*]] = or i64 [[OR]], [[CONV2]]
+; CHECK-NEXT:    ret i64 [[OR2]]
+;
+  %conv = zext i16 %b to i64
+  %conv2 = zext i16 %c to i64
+  %or = or i64 %a, %conv
+  %or2 = or i64 %or, %conv2
+  ret i64 %or2
+}
+
+define i64 @sext_and_chain(i64 %a, i16 %b, i16 %c) {
+; CHECK-LABEL: @sext_and_chain(
+; CHECK-NEXT:    [[CONV:%.*]] = sext i16 [[B:%.*]] to i64
+; CHECK-NEXT:    [[CONV2:%.*]] = sext i16 [[C:%.*]] to i64
+; CHECK-NEXT:    [[AND:%.*]] = and i64 [[CONV]], [[A:%.*]]
+; CHECK-NEXT:    [[AND2:%.*]] = and i64 [[AND]], [[CONV2]]
+; CHECK-NEXT:    ret i64 [[AND2]]
+;
+  %conv = sext i16 %b to i64
+  %conv2 = sext i16 %c to i64
+  %and = and i64 %a, %conv
+  %and2 = and i64 %and, %conv2
+  ret i64 %and2
+}
+
+define i64 @zext_and_chain(i64 %a, i16 %b, i16 %c) {
+; CHECK-LABEL: @zext_and_chain(
+; CHECK-NEXT:    [[CONV:%.*]] = zext i16 [[B:%.*]] to i64
+; CHECK-NEXT:    [[CONV2:%.*]] = zext i16 [[C:%.*]] to i64
+; CHECK-NEXT:    [[AND:%.*]] = and i64 [[CONV]], [[A:%.*]]
+; CHECK-NEXT:    [[AND2:%.*]] = and i64 [[AND]], [[CONV2]]
+; CHECK-NEXT:    ret i64 [[AND2]]
+;
+  %conv = zext i16 %b to i64
+  %conv2 = zext i16 %c to i64
+  %and = and i64 %a, %conv
+  %and2 = and i64 %and, %conv2
+  ret i64 %and2
+}
+
+define i64 @sext_xor_chain(i64 %a, i16 %b, i16 %c) {
+; CHECK-LABEL: @sext_xor_chain(
+; CHECK-NEXT:    [[CONV:%.*]] = sext i16 [[B:%.*]] to i64
+; CHECK-NEXT:    [[CONV2:%.*]] = sext i16 [[C:%.*]] to i64
+; CHECK-NEXT:    [[XOR:%.*]] = xor i64 [[CONV]], [[A:%.*]]
+; CHECK-NEXT:    [[XOR2:%.*]] = xor i64 [[XOR]], [[CONV2]]
+; CHECK-NEXT:    ret i64 [[XOR2]]
+;
+  %conv = sext i16 %b to i64
+  %conv2 = sext i16 %c to i64
+  %xor = xor i64 %a, %conv
+  %xor2 = xor i64 %xor, %conv2
+  ret i64 %xor2
+}
+
+define i64 @zext_xor_chain(i64 %a, i16 %b, i16 %c) {
+; CHECK-LABEL: @zext_xor_chain(
+; CHECK-NEXT:    [[CONV:%.*]] = zext i16 [[B:%.*]] to i64
+; CHECK-NEXT:    [[CONV2:%.*]] = zext i16 [[C:%.*]] to i64
+; CHECK-NEXT:    [[XOR:%.*]] = xor i64 [[CONV]], [[A:%.*]]
+; CHECK-NEXT:    [[XOR2:%.*]] = xor i64 [[XOR]], [[CONV2]]
+; CHECK-NEXT:    ret i64 [[XOR2]]
+;
+  %conv = zext i16 %b to i64
+  %conv2 = zext i16 %c to i64
+  %xor = xor i64 %a, %conv
+  %xor2 = xor i64 %xor, %conv2
+  ret i64 %xor2
+}
+
+; Negative test with more uses.
+define i64 @sext_or_chain_two_uses1(i64 %a, i16 %b, i16 %c, i64 %d) {
+; CHECK-LABEL: @sext_or_chain_two_uses1(
+; CHECK-NEXT:    [[CONV:%.*]] = sext i16 [[B:%.*]] to i64
+; CHECK-NEXT:    [[CONV2:%.*]] = sext i16 [[C:%.*]] to i64
+; CHECK-NEXT:    [[OR:%.*]] = or i64 [[CONV]], [[A:%.*]]
+; CHECK-NEXT:    [[OR2:%.*]] = or i64 [[OR]], [[CONV2]]
+; CHECK-NEXT:    [[USE:%.*]] = udiv i64 [[OR]], [[D:%.*]]
+; CHECK-NEXT:    [[RETVAL:%.*]] = udiv i64 [[OR2]], [[USE]]
+; CHECK-NEXT:    ret i64 [[RETVAL]]
+;
+  %conv = sext i16 %b to i64
+  %conv2 = sext i16 %c to i64
+  ; %or has two uses
+  %or = or i64 %a, %conv
+  %or2 = or i64 %or, %conv2
+  %use = udiv i64 %or, %d
+  %retval = udiv i64 %or2, %use
+  ret i64 %retval
+}
+define i64 @sext_or_chain_two_uses2(i64 %a, i16 %b, i16 %c, i64 %d) {
+; CHECK-LABEL: @sext_or_chain_two_uses2(
+; CHECK-NEXT:    [[CONV:%.*]] = sext i16 [[B:%.*]] to i64
+; CHECK-NEXT:    [[CONV2:%.*]] = sext i16 [[C:%.*]] to i64
+; CHECK-NEXT:    [[OR:%.*]] = or i64 [[CONV]], [[A:%.*]]
+; CHECK-NEXT:    [[OR2:%.*]] = or i64 [[OR]], [[CONV2]]
+; CHECK-NEXT:    [[USE1:%.*]] = udiv i64 [[OR2]], [[D:%.*]]
+; CHECK-NEXT:    [[USE2:%.*]] = udiv i64 [[OR2]], [[USE1]]
+; CHECK-NEXT:    ret i64 [[USE2]]
+;
+  %conv = sext i16 %b to i64
+  %conv2 = sext i16 %c to i64
+  %or = or i64 %a, %conv
+  ; %or2 has two uses
+  %or2 = or i64 %or, %conv2
+  %use1 = udiv i64 %or2, %d
+  %use2 = udiv i64 %or2, %use1
+  ret i64 %use2
+}
-- 
2.7.4