llvm/test/Transforms/InstCombine/sub-lshr-or-to-icmp-select.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt -instcombine %s -S -o - | FileCheck %s
   3
   4 ; Scalar Types
   5
   6 define i32 @neg_or_lshr_i32(i32 %x) {
   7 ; CHECK-LABEL: @neg_or_lshr_i32(
   8 ; CHECK-NEXT:    [[TMP1:%.*]] = icmp ne i32 [[X:%.*]], 0
   9 ; CHECK-NEXT:    [[SHR:%.*]] = zext i1 [[TMP1]] to i32
  10 ; CHECK-NEXT:    ret i32 [[SHR]]
  11 ;
  12   %neg = sub i32 0, %x
  13   %or = or i32 %neg, %x
  14   %shr = lshr i32 %or, 31
  15   ret i32 %shr
  16 }
  17
  18 ; Commute
  19
  20 define i32 @neg_or_lshr_i32_commute(i32 %x0) {
  21 ; CHECK-LABEL: @neg_or_lshr_i32_commute(
  22 ; CHECK-NEXT:    [[X:%.*]] = sdiv i32 42, [[X0:%.*]]
  23 ; CHECK-NEXT:    [[TMP1:%.*]] = icmp ne i32 [[X]], 0
  24 ; CHECK-NEXT:    [[SHR:%.*]] = zext i1 [[TMP1]] to i32
  25 ; CHECK-NEXT:    ret i32 [[SHR]]
  26 ;
  27   %x = sdiv i32 42, %x0 ; thwart complexity-based canonicalization
  28   %neg = sub i32 0, %x
  29   %or = or i32 %x, %neg
  30   %shr = lshr i32 %or, 31
  31   ret i32 %shr
  32 }
  33
  34 ; Vector Types
  35
  36 define <4 x i32> @neg_or_lshr_i32_vec(<4 x i32> %x) {
  37 ; CHECK-LABEL: @neg_or_lshr_i32_vec(
  38 ; CHECK-NEXT:    [[TMP1:%.*]] = icmp ne <4 x i32> [[X:%.*]], zeroinitializer
  39 ; CHECK-NEXT:    [[SHR:%.*]] = zext <4 x i1> [[TMP1]] to <4 x i32>
  40 ; CHECK-NEXT:    ret <4 x i32> [[SHR]]
  41 ;
  42   %neg = sub <4 x i32> zeroinitializer, %x
  43   %or = or <4 x i32> %neg, %x
  44   %shr = lshr <4 x i32> %or, <i32 31, i32 31, i32 31, i32 31>
  45   ret <4 x i32> %shr
  46 }
  47
  48 define <4 x i32> @neg_or_lshr_i32_vec_commute(<4 x i32> %x0) {
  49 ; CHECK-LABEL: @neg_or_lshr_i32_vec_commute(
  50 ; CHECK-NEXT:    [[X:%.*]] = sdiv <4 x i32> <i32 42, i32 42, i32 42, i32 42>, [[X0:%.*]]
  51 ; CHECK-NEXT:    [[TMP1:%.*]] = icmp ne <4 x i32> [[X]], zeroinitializer
  52 ; CHECK-NEXT:    [[SHR:%.*]] = zext <4 x i1> [[TMP1]] to <4 x i32>
  53 ; CHECK-NEXT:    ret <4 x i32> [[SHR]]
  54 ;
  55   %x = sdiv <4 x i32> <i32 42, i32 42, i32 42, i32 42>, %x0 ; thwart complexity-based canonicalization
  56   %neg = sub <4 x i32> zeroinitializer, %x
  57   %or = or <4 x i32> %x, %neg
  58   %shr = lshr <4 x i32> %or, <i32 31, i32 31, i32 31, i32 31>
  59   ret <4 x i32> %shr
  60 }
  61
  62 ; Extra uses
  63
  64 define i32 @neg_extra_use_or_lshr_i32(i32 %x, i32* %p) {
  65 ; CHECK-LABEL: @neg_extra_use_or_lshr_i32(
  66 ; CHECK-NEXT:    [[NEG:%.*]] = sub i32 0, [[X:%.*]]
  67 ; CHECK-NEXT:    [[TMP1:%.*]] = icmp ne i32 [[X]], 0
  68 ; CHECK-NEXT:    [[SHR:%.*]] = zext i1 [[TMP1]] to i32
  69 ; CHECK-NEXT:    store i32 [[NEG]], i32* [[P:%.*]], align 4
  70 ; CHECK-NEXT:    ret i32 [[SHR]]
  71 ;
  72   %neg = sub i32 0, %x
  73   %or = or i32 %neg, %x
  74   %shr = lshr i32 %or, 31
  75   store i32 %neg, i32* %p
  76   ret i32 %shr
  77 }
  78
  79 ; Negative Tests
  80
  81 define i32 @neg_or_extra_use_lshr_i32(i32 %x, i32* %p) {
  82 ; CHECK-LABEL: @neg_or_extra_use_lshr_i32(
  83 ; CHECK-NEXT:    [[NEG:%.*]] = sub i32 0, [[X:%.*]]
  84 ; CHECK-NEXT:    [[OR:%.*]] = or i32 [[NEG]], [[X]]
  85 ; CHECK-NEXT:    [[SHR:%.*]] = lshr i32 [[OR]], 31
  86 ; CHECK-NEXT:    store i32 [[OR]], i32* [[P:%.*]], align 4
  87 ; CHECK-NEXT:    ret i32 [[SHR]]
  88 ;
  89   %neg = sub i32 0, %x
  90   %or = or i32 %neg, %x
  91   %shr = lshr i32 %or, 31
  92   store i32 %or, i32* %p
  93   ret i32 %shr
  94 }