llvm/test/Transforms/InstCombine/sdiv-exact-by-negative-power-of-two.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt < %s -passes=instcombine -S | FileCheck %s
   3
   4 ; Fold
   5 ;   x s/EXACT (-1 << y)
   6 ; to
   7 ;   -(x a>>EXACT y)
   8
   9 define i8 @t0(i8 %x) {
  10 ; CHECK-LABEL: @t0(
  11 ; CHECK-NEXT:    [[DIV_NEG:%.*]] = ashr exact i8 [[X:%.*]], 5
  12 ; CHECK-NEXT:    [[DIV:%.*]] = sub nsw i8 0, [[DIV_NEG]]
  13 ; CHECK-NEXT:    ret i8 [[DIV]]
  14 ;
  15   %div = sdiv exact i8 %x, -32
  16   ret i8 %div
  17 }
  18
  19 define i8 @n1(i8 %x) {
  20 ; CHECK-LABEL: @n1(
  21 ; CHECK-NEXT:    [[DIV:%.*]] = sdiv i8 [[X:%.*]], -32
  22 ; CHECK-NEXT:    ret i8 [[DIV]]
  23 ;
  24   %div = sdiv i8 %x, -32 ; not exact
  25   ret i8 %div
  26 }
  27
  28 define <2 x i8> @t2_vec_splat(<2 x i8> %x) {
  29 ; CHECK-LABEL: @t2_vec_splat(
  30 ; CHECK-NEXT:    [[DIV_NEG:%.*]] = ashr exact <2 x i8> [[X:%.*]], splat (i8 5)
  31 ; CHECK-NEXT:    [[DIV:%.*]] = sub nsw <2 x i8> zeroinitializer, [[DIV_NEG]]
  32 ; CHECK-NEXT:    ret <2 x i8> [[DIV]]
  33 ;
  34   %div = sdiv exact <2 x i8> %x, <i8 -32, i8 -32>
  35   ret <2 x i8> %div
  36 }
  37
  38 define <2 x i8> @t3_vec(<2 x i8> %x) {
  39 ; CHECK-LABEL: @t3_vec(
  40 ; CHECK-NEXT:    [[DIV_NEG:%.*]] = ashr exact <2 x i8> [[X:%.*]], <i8 5, i8 4>
  41 ; CHECK-NEXT:    [[DIV:%.*]] = sub nsw <2 x i8> zeroinitializer, [[DIV_NEG]]
  42 ; CHECK-NEXT:    ret <2 x i8> [[DIV]]
  43 ;
  44   %div = sdiv exact <2 x i8> %x, <i8 -32, i8 -16>
  45   ret <2 x i8> %div
  46 }
  47
  48 define <2 x i8> @n4_vec_mixed(<2 x i8> %x) {
  49 ; CHECK-LABEL: @n4_vec_mixed(
  50 ; CHECK-NEXT:    [[DIV:%.*]] = sdiv exact <2 x i8> [[X:%.*]], <i8 -32, i8 16>
  51 ; CHECK-NEXT:    ret <2 x i8> [[DIV]]
  52 ;
  53   %div = sdiv exact <2 x i8> %x, <i8 -32, i8 16>
  54   ret <2 x i8> %div
  55 }
  56
  57 define <2 x i8> @n4_vec_undef(<2 x i8> %x) {
  58 ; CHECK-LABEL: @n4_vec_undef(
  59 ; CHECK-NEXT:    ret <2 x i8> poison
  60 ;
  61   %div = sdiv exact <2 x i8> %x, <i8 -32, i8 undef>
  62   ret <2 x i8> %div
  63 }
  64
  65 define i8 @prove_exact_with_high_mask(i8 %x, i8 %y) {
  66 ; CHECK-LABEL: @prove_exact_with_high_mask(
  67 ; CHECK-NEXT:    [[A:%.*]] = ashr i8 [[X:%.*]], 2
  68 ; CHECK-NEXT:    [[D_NEG:%.*]] = and i8 [[A]], -8
  69 ; CHECK-NEXT:    [[D:%.*]] = sub nsw i8 0, [[D_NEG]]
  70 ; CHECK-NEXT:    ret i8 [[D]]
  71 ;
  72   %a = and i8 %x, -32
  73   %d = sdiv i8 %a, -4
  74   ret i8 %d
  75 }
  76
  77 define i8 @prove_exact_with_high_mask_limit(i8 %x, i8 %y) {
  78 ; CHECK-LABEL: @prove_exact_with_high_mask_limit(
  79 ; CHECK-NEXT:    [[D_NEG:%.*]] = ashr i8 [[X:%.*]], 5
  80 ; CHECK-NEXT:    [[D:%.*]] = sub nsw i8 0, [[D_NEG]]
  81 ; CHECK-NEXT:    ret i8 [[D]]
  82 ;
  83   %a = and i8 %x, -32
  84   %d = sdiv i8 %a, -32
  85   ret i8 %d
  86 }
  87
  88 ; negative test - not enough low zeros in dividend
  89
  90 define i8 @not_prove_exact_with_high_mask(i8 %x, i8 %y) {
  91 ; CHECK-LABEL: @not_prove_exact_with_high_mask(
  92 ; CHECK-NEXT:    [[A:%.*]] = and i8 [[X:%.*]], -32
  93 ; CHECK-NEXT:    [[D:%.*]] = sdiv i8 [[A]], -64
  94 ; CHECK-NEXT:    ret i8 [[D]]
  95 ;
  96   %a = and i8 %x, -32
  97   %d = sdiv i8 %a, -64
  98   ret i8 %d
  99 }
 100
 101 define <2 x i8> @prove_exact_with_high_mask_splat_vec(<2 x i8> %x, <2 x i8> %y) {
 102 ; CHECK-LABEL: @prove_exact_with_high_mask_splat_vec(
 103 ; CHECK-NEXT:    [[A:%.*]] = shl <2 x i8> [[X:%.*]], splat (i8 5)
 104 ; CHECK-NEXT:    [[D_NEG:%.*]] = ashr exact <2 x i8> [[A]], splat (i8 4)
 105 ; CHECK-NEXT:    [[D:%.*]] = sub nsw <2 x i8> zeroinitializer, [[D_NEG]]
 106 ; CHECK-NEXT:    ret <2 x i8> [[D]]
 107 ;
 108   %a = shl <2 x i8> %x, <i8 5, i8 5>
 109   %d = sdiv <2 x i8> %a, <i8 -16, i8 -16>
 110   ret <2 x i8> %d
 111 }
 112
 113 ; TODO: Needs knownbits to handle arbitrary vector constants.
 114
 115 define <2 x i8> @prove_exact_with_high_mask_vec(<2 x i8> %x, <2 x i8> %y) {
 116 ; CHECK-LABEL: @prove_exact_with_high_mask_vec(
 117 ; CHECK-NEXT:    [[A:%.*]] = shl <2 x i8> [[X:%.*]], <i8 3, i8 2>
 118 ; CHECK-NEXT:    [[D:%.*]] = sdiv <2 x i8> [[A]], <i8 -8, i8 -4>
 119 ; CHECK-NEXT:    ret <2 x i8> [[D]]
 120 ;
 121   %a = shl <2 x i8> %x, <i8 3, i8 2>
 122   %d = sdiv <2 x i8> %a, <i8 -8, i8 -4>
 123   ret <2 x i8> %d
 124 }