llvm/test/Transforms/Reassociate/wrap-flags.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt < %s -reassociate -dce -S | FileCheck %s
   3 ; PR12985
   4
   5 ; Verify the nsw flags are preserved when converting shl to mul.
   6
   7 define i32 @shl_to_mul_nsw(i32 %i) {
   8 ;
   9 ; CHECK-LABEL: @shl_to_mul_nsw(
  10 ; CHECK-NEXT:  entry:
  11 ; CHECK-NEXT:    [[MUL:%.*]] = mul i32 [[I:%.*]], -2147483648
  12 ; CHECK-NEXT:    [[MUL2:%.*]] = add i32 [[MUL]], 1
  13 ; CHECK-NEXT:    ret i32 [[MUL2]]
  14 ;
  15 entry:
  16   %mul = shl nsw i32 %i, 31
  17   %mul2 = add i32 %mul, 1
  18   ret i32 %mul2
  19 }
  20
  21 define i32 @shl_to_mul_nsw_2(i32 %i) {
  22 ;
  23 ; CHECK-LABEL: @shl_to_mul_nsw_2(
  24 ; CHECK-NEXT:  entry:
  25 ; CHECK-NEXT:    [[MUL:%.*]] = mul nsw i32 [[I:%.*]], 1073741824
  26 ; CHECK-NEXT:    [[MUL2:%.*]] = add i32 [[MUL]], 1
  27 ; CHECK-NEXT:    ret i32 [[MUL2]]
  28 ;
  29 entry:
  30   %mul = shl nsw i32 %i, 30
  31   %mul2 = add i32 %mul, 1
  32   ret i32 %mul2
  33 }
  34
  35 define i32 @shl_to_mul_nuw(i32 %i) {
  36 ;
  37 ; CHECK-LABEL: @shl_to_mul_nuw(
  38 ; CHECK-NEXT:  entry:
  39 ; CHECK-NEXT:    [[MUL:%.*]] = mul nuw i32 [[I:%.*]], 4
  40 ; CHECK-NEXT:    [[MUL2:%.*]] = add i32 [[MUL]], 1
  41 ; CHECK-NEXT:    ret i32 [[MUL2]]
  42 ;
  43 entry:
  44   %mul = shl nuw i32 %i, 2
  45   %mul2 = add i32 %mul, 1
  46   ret i32 %mul2
  47 }
  48
  49 define i32 @shl_to_mul_nuw_nsw(i32 %i) {
  50 ;
  51 ; CHECK-LABEL: @shl_to_mul_nuw_nsw(
  52 ; CHECK-NEXT:  entry:
  53 ; CHECK-NEXT:    [[MUL:%.*]] = mul nuw nsw i32 [[I:%.*]], 4
  54 ; CHECK-NEXT:    [[MUL2:%.*]] = add i32 [[MUL]], 1
  55 ; CHECK-NEXT:    ret i32 [[MUL2]]
  56 ;
  57 entry:
  58   %mul = shl nuw nsw i32 %i, 2
  59   %mul2 = add i32 %mul, 1
  60   ret i32 %mul2
  61 }
  62
  63 define i32 @shl_to_mul_nuw_nsw_bitwidth_m1(i32 %i) {
  64 ;
  65 ; CHECK-LABEL: @shl_to_mul_nuw_nsw_bitwidth_m1(
  66 ; CHECK-NEXT:  entry:
  67 ; CHECK-NEXT:    [[MUL:%.*]] = mul nuw nsw i32 [[I:%.*]], -2147483648
  68 ; CHECK-NEXT:    [[MUL2:%.*]] = add i32 [[MUL]], 1
  69 ; CHECK-NEXT:    ret i32 [[MUL2]]
  70 ;
  71 entry:
  72   %mul = shl nuw nsw i32 %i, 31
  73   %mul2 = add i32 %mul, 1
  74   ret i32 %mul2
  75 }
  76
  77 define i2 @pr23926(i2 %X1, i2 %X2) {
  78 ;
  79 ; CHECK-LABEL: @pr23926(
  80 ; CHECK-NEXT:    [[X1_NEG:%.*]] = sub i2 0, [[X1:%.*]]
  81 ; CHECK-NEXT:    [[ADD_NEG:%.*]] = add i2 [[X1_NEG]], -1
  82 ; CHECK-NEXT:    [[SUB:%.*]] = add i2 [[ADD_NEG]], [[X2:%.*]]
  83 ; CHECK-NEXT:    ret i2 [[SUB]]
  84 ;
  85   %add = add nuw i2 %X1, 1
  86   %sub = sub nuw nsw i2 %X2, %add
  87   ret i2 %sub
  88 }