llvm/test/Transforms/InstCombine/nsw-inseltpoison.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt < %s -passes=instcombine -S | FileCheck %s
   3
   4 define i32 @sub1(i32 %x) {
   5 ; CHECK-LABEL: @sub1(
   6 ; CHECK-NEXT:    [[Y:%.*]] = sub i32 0, [[X:%.*]]
   7 ; CHECK-NEXT:    [[Z:%.*]] = sdiv i32 [[Y]], 337
   8 ; CHECK-NEXT:    ret i32 [[Z]]
   9 ;
  10   %y = sub i32 0, %x
  11   %z = sdiv i32 %y, 337
  12   ret i32 %z
  13 }
  14
  15 define i32 @sub2(i32 %x) {
  16 ; CHECK-LABEL: @sub2(
  17 ; CHECK-NEXT:    [[Z:%.*]] = sdiv i32 [[X:%.*]], -337
  18 ; CHECK-NEXT:    ret i32 [[Z]]
  19 ;
  20   %y = sub nsw i32 0, %x
  21   %z = sdiv i32 %y, 337
  22   ret i32 %z
  23 }
  24
  25 define i1 @shl_icmp(i64 %X) {
  26 ; CHECK-LABEL: @shl_icmp(
  27 ; CHECK-NEXT:    [[B:%.*]] = icmp eq i64 [[X:%.*]], 0
  28 ; CHECK-NEXT:    ret i1 [[B]]
  29 ;
  30   %A = shl nuw i64 %X, 2   ; X/4
  31   %B = icmp eq i64 %A, 0
  32   ret i1 %B
  33 }
  34
  35 define i64 @shl1(i64 %X, ptr %P) {
  36 ; CHECK-LABEL: @shl1(
  37 ; CHECK-NEXT:    [[A:%.*]] = and i64 [[X:%.*]], 312
  38 ; CHECK-NEXT:    store i64 [[A]], ptr [[P:%.*]], align 4
  39 ; CHECK-NEXT:    [[B:%.*]] = shl nuw nsw i64 [[A]], 8
  40 ; CHECK-NEXT:    ret i64 [[B]]
  41 ;
  42   %A = and i64 %X, 312
  43   store i64 %A, ptr %P  ; multiple uses of A.
  44   %B = shl i64 %A, 8
  45   ret i64 %B
  46 }
  47
  48 define i32 @preserve1(i32 %x) {
  49 ; CHECK-LABEL: @preserve1(
  50 ; CHECK-NEXT:    [[ADD3:%.*]] = add nsw i32 [[X:%.*]], 5
  51 ; CHECK-NEXT:    ret i32 [[ADD3]]
  52 ;
  53   %add = add nsw i32 %x, 2
  54   %add3 = add nsw i32 %add, 3
  55   ret i32 %add3
  56 }
  57
  58 define i8 @nopreserve1(i8 %x) {
  59 ; CHECK-LABEL: @nopreserve1(
  60 ; CHECK-NEXT:    [[ADD3:%.*]] = add i8 [[X:%.*]], -126
  61 ; CHECK-NEXT:    ret i8 [[ADD3]]
  62 ;
  63   %add = add nsw i8 %x, 127
  64   %add3 = add nsw i8 %add, 3
  65   ret i8 %add3
  66 }
  67
  68 define i8 @nopreserve2(i8 %x) {
  69 ; CHECK-LABEL: @nopreserve2(
  70 ; CHECK-NEXT:    [[ADD3:%.*]] = add i8 [[X:%.*]], 3
  71 ; CHECK-NEXT:    ret i8 [[ADD3]]
  72 ;
  73   %add = add i8 %x, 1
  74   %add3 = add nsw i8 %add, 2
  75   ret i8 %add3
  76 }
  77
  78 define i8 @nopreserve3(i8 %A, i8 %B) {
  79 ; CHECK-LABEL: @nopreserve3(
  80 ; CHECK-NEXT:    [[Y:%.*]] = add i8 [[A:%.*]], [[B:%.*]]
  81 ; CHECK-NEXT:    [[ADD:%.*]] = add i8 [[Y]], 20
  82 ; CHECK-NEXT:    ret i8 [[ADD]]
  83 ;
  84   %x = add i8 %A, 10
  85   %y = add i8 %B, 10
  86   %add = add nsw i8 %x, %y
  87   ret i8 %add
  88 }
  89
  90 define i8 @nopreserve4(i8 %A, i8 %B) {
  91 ; CHECK-LABEL: @nopreserve4(
  92 ; CHECK-NEXT:    [[Y:%.*]] = add i8 [[A:%.*]], [[B:%.*]]
  93 ; CHECK-NEXT:    [[ADD:%.*]] = add i8 [[Y]], 20
  94 ; CHECK-NEXT:    ret i8 [[ADD]]
  95 ;
  96   %x = add nsw i8 %A, 10
  97   %y = add nsw i8 %B, 10
  98   %add = add nsw i8 %x, %y
  99   ret i8 %add
 100 }
 101
 102 define <3 x i32> @shl_nuw_nsw_shuffle_splat_vec(<2 x i8> %x) {
 103 ; CHECK-LABEL: @shl_nuw_nsw_shuffle_splat_vec(
 104 ; CHECK-NEXT:    [[T2:%.*]] = zext <2 x i8> [[X:%.*]] to <2 x i32>
 105 ; CHECK-NEXT:    [[SHUF:%.*]] = shufflevector <2 x i32> [[T2]], <2 x i32> poison, <3 x i32> <i32 1, i32 0, i32 1>
 106 ; CHECK-NEXT:    [[T3:%.*]] = shl nuw nsw <3 x i32> [[SHUF]], splat (i32 17)
 107 ; CHECK-NEXT:    ret <3 x i32> [[T3]]
 108 ;
 109   %t2 = zext <2 x i8> %x to <2 x i32>
 110   %shuf = shufflevector <2 x i32> %t2, <2 x i32> poison, <3 x i32> <i32 1, i32 0, i32 1>
 111   %t3 = shl <3 x i32> %shuf, <i32 17, i32 17, i32 17>
 112   ret <3 x i32> %t3
 113 }
 114
 115 ; Negative test - if the shuffle mask contains an undef, we bail out to
 116 ; avoid propagating information that may not be used consistently by callers.
 117
 118 define <3 x i32> @shl_nuw_nsw_shuffle_undef_elt_splat_vec(<2 x i8> %x) {
 119 ; CHECK-LABEL: @shl_nuw_nsw_shuffle_undef_elt_splat_vec(
 120 ; CHECK-NEXT:    [[T2:%.*]] = zext <2 x i8> [[X:%.*]] to <2 x i32>
 121 ; CHECK-NEXT:    [[SHUF:%.*]] = shufflevector <2 x i32> [[T2]], <2 x i32> poison, <3 x i32> <i32 1, i32 poison, i32 0>
 122 ; CHECK-NEXT:    [[T3:%.*]] = shl <3 x i32> [[SHUF]], splat (i32 17)
 123 ; CHECK-NEXT:    ret <3 x i32> [[T3]]
 124 ;
 125   %t2 = zext <2 x i8> %x to <2 x i32>
 126   %shuf = shufflevector <2 x i32> %t2, <2 x i32> poison, <3 x i32> <i32 1, i32 undef, i32 0>
 127   %t3 = shl <3 x i32> %shuf, <i32 17, i32 17, i32 17>
 128   ret <3 x i32> %t3
 129 }
 130
 131 ; Make sure we don't crash on a ConstantExpr shufflevector
 132 define <vscale x 2 x i64> @mul_nuw_nsw_shuffle_constant_expr(<vscale x 2 x i8> %z) {
 133 ; CHECK-LABEL: @mul_nuw_nsw_shuffle_constant_expr(
 134 ; CHECK-NEXT:    [[XX:%.*]] = zext <vscale x 2 x i8> [[Z:%.*]] to <vscale x 2 x i64>
 135 ; CHECK-NEXT:    [[T3:%.*]] = mul <vscale x 2 x i64> [[XX]], shufflevector (<vscale x 2 x i64> insertelement (<vscale x 2 x i64> undef, i64 3, i32 0), <vscale x 2 x i64> zeroinitializer, <vscale x 2 x i32> zeroinitializer)
 136 ; CHECK-NEXT:    ret <vscale x 2 x i64> [[T3]]
 137 ;
 138   %xx = zext <vscale x 2 x i8> %z to <vscale x 2 x i64>
 139   %shuf = shufflevector <vscale x 2 x i64> insertelement (<vscale x 2 x i64> undef, i64 3, i32 0), <vscale x 2 x i64> zeroinitializer, <vscale x 2 x i32> zeroinitializer
 140   %t3 = mul <vscale x 2 x i64> %shuf, %xx
 141   ret <vscale x 2 x i64> %t3
 142 }