From 7bc54cebeab437eaeb5fb39c876f9d24e38f6f52 Mon Sep 17 00:00:00 2001
From: Michael Kuperstein <mkuper@google.com>
Date: Thu, 1 Sep 2016 23:02:32 +0000
Subject: [PATCH] [Legalizer] Don't throw away false low half when expanding
 GT/LT SETCC

When expanding a SETCC for which the low half is known to evaluate to false,
we can only throw it away for LT/GT comparisons, not LE/GE.

This fixes PR29170.

Differential Revision: https://reviews.llvm.org/D24151

llvm-svn: 280424
---
 .../CodeGen/SelectionDAG/LegalizeIntegerTypes.cpp  | 18 ++++++------
 llvm/test/CodeGen/X86/pr29170.ll                   | 32 ++++++++++++++++++++++
 2 files changed, 41 insertions(+), 9 deletions(-)
 create mode 100644 llvm/test/CodeGen/X86/pr29170.ll

diff --git a/llvm/lib/CodeGen/SelectionDAG/LegalizeIntegerTypes.cpp b/llvm/lib/CodeGen/SelectionDAG/LegalizeIntegerTypes.cpp
index 47be5f2..5b1fbbb 100644
--- a/llvm/lib/CodeGen/SelectionDAG/LegalizeIntegerTypes.cpp
+++ b/llvm/lib/CodeGen/SelectionDAG/LegalizeIntegerTypes.cpp
@@ -2865,16 +2865,16 @@ void DAGTypeLegalizer::IntegerExpandSetCCOperands(SDValue &NewLHS,
 
   ConstantSDNode *LoCmpC = dyn_cast<ConstantSDNode>(LoCmp.getNode());
   ConstantSDNode *HiCmpC = dyn_cast<ConstantSDNode>(HiCmp.getNode());
-  if ((LoCmpC && LoCmpC->isNullValue()) ||
-      (HiCmpC && HiCmpC->isNullValue() &&
-       (CCCode == ISD::SETLE || CCCode == ISD::SETGE || CCCode == ISD::SETUGE ||
-        CCCode == ISD::SETULE)) ||
-      (HiCmpC && HiCmpC->getAPIntValue() == 1 &&
-       (CCCode == ISD::SETLT || CCCode == ISD::SETGT || CCCode == ISD::SETUGT ||
-        CCCode == ISD::SETULT))) {
-    // low part is known false, returns high part.
+
+  bool EqAllowed = (CCCode == ISD::SETLE || CCCode == ISD::SETGE ||
+                    CCCode == ISD::SETUGE || CCCode == ISD::SETULE);
+
+  if ((EqAllowed && (HiCmpC && HiCmpC->isNullValue())) ||
+      (!EqAllowed && ((HiCmpC && (HiCmpC->getAPIntValue() == 1)) ||
+                      (LoCmpC && LoCmpC->isNullValue())))) {
     // For LE / GE, if high part is known false, ignore the low part.
-    // For LT / GT, if high part is known true, ignore the low part.
+    // For LT / GT: if low part is known false, return the high part.
+    //              if high part is known true, ignore the low part.
     NewLHS = HiCmp;
     NewRHS = SDValue();
     return;
diff --git a/llvm/test/CodeGen/X86/pr29170.ll b/llvm/test/CodeGen/X86/pr29170.ll
new file mode 100644
index 0000000..d8e2755
--- /dev/null
+++ b/llvm/test/CodeGen/X86/pr29170.ll
@@ -0,0 +1,32 @@
+; RUN: llc < %s | FileCheck %s
+
+target datalayout = "e-m:e-p:32:32-f64:32:64-f80:32-n8:16:32-S128"
+target triple = "i386-unknown-linux-gnu"
+
+@b = global i16 0, align 4
+
+; CHECK-LABEL: @main
+; CHECK: cmpl
+; CHECK: sbbl
+define i32 @main() {
+entry:
+  %true = icmp eq i32 0, 0
+  %const = bitcast i64 -4294967296 to i64
+  br i1 %true, label %go, label %if.else
+
+go:
+  %b = load i16, i16* @b, align 4
+  %sext = shl i16 %b, 8
+  %conv = ashr i16 %sext, 8
+  %neg4 = xor i64 %const, -1
+  %conv5 = zext i16 %conv to i64
+  %cmp = icmp slt i64 %conv5, %neg4
+  br i1 %cmp, label %if.then, label %if.else
+
+if.then:
+  ret i32 42
+
+if.else:
+  ret i32 0
+}
+
-- 
2.7.4