llvm/test/Transforms/InstSimplify/abs_intrinsic.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt < %s -instsimplify -S | FileCheck %s
   3
   4 declare i8 @llvm.abs.i8(i8, i1)
   5 declare i32 @llvm.abs.i32(i32, i1)
   6 declare <3 x i82> @llvm.abs.v3i82(<3 x i82>, i1)
   7 declare <4 x i32> @llvm.abs.v4i32(<4 x i32>, i1)
   8
   9 define i32 @test_abs_abs_0(i32 %x) {
  10 ; CHECK-LABEL: @test_abs_abs_0(
  11 ; CHECK-NEXT:    [[A:%.*]] = call i32 @llvm.abs.i32(i32 [[X:%.*]], i1 false)
  12 ; CHECK-NEXT:    ret i32 [[A]]
  13 ;
  14   %a = call i32 @llvm.abs.i32(i32 %x, i1 false)
  15   %b = call i32 @llvm.abs.i32(i32 %a, i1 false)
  16   ret i32 %b
  17 }
  18
  19 define i32 @test_abs_abs_1(i32 %x) {
  20 ; CHECK-LABEL: @test_abs_abs_1(
  21 ; CHECK-NEXT:    [[A:%.*]] = call i32 @llvm.abs.i32(i32 [[X:%.*]], i1 true)
  22 ; CHECK-NEXT:    ret i32 [[A]]
  23 ;
  24   %a = call i32 @llvm.abs.i32(i32 %x, i1 true)
  25   %b = call i32 @llvm.abs.i32(i32 %a, i1 false)
  26   ret i32 %b
  27 }
  28
  29 define i32 @test_abs_abs_2(i32 %x) {
  30 ; CHECK-LABEL: @test_abs_abs_2(
  31 ; CHECK-NEXT:    [[A:%.*]] = call i32 @llvm.abs.i32(i32 [[X:%.*]], i1 false)
  32 ; CHECK-NEXT:    ret i32 [[A]]
  33 ;
  34   %a = call i32 @llvm.abs.i32(i32 %x, i1 false)
  35   %b = call i32 @llvm.abs.i32(i32 %a, i1 true)
  36   ret i32 %b
  37 }
  38
  39 define i32 @test_abs_abs_3(i32 %x) {
  40 ; CHECK-LABEL: @test_abs_abs_3(
  41 ; CHECK-NEXT:    [[A:%.*]] = call i32 @llvm.abs.i32(i32 [[X:%.*]], i1 true)
  42 ; CHECK-NEXT:    ret i32 [[A]]
  43 ;
  44   %a = call i32 @llvm.abs.i32(i32 %x, i1 true)
  45   %b = call i32 @llvm.abs.i32(i32 %a, i1 true)
  46   ret i32 %b
  47 }
  48
  49 ; If the sign bit is known zero, the abs is not needed.
  50 ; These cases are only folded by InstCombine, to avoid computing known bits
  51 ; twice, for the non-negative and the negative case.
  52
  53 define i32 @zext_abs(i31 %x) {
  54 ; CHECK-LABEL: @zext_abs(
  55 ; CHECK-NEXT:    [[ZEXT:%.*]] = zext i31 [[X:%.*]] to i32
  56 ; CHECK-NEXT:    [[ABS:%.*]] = call i32 @llvm.abs.i32(i32 [[ZEXT]], i1 false)
  57 ; CHECK-NEXT:    ret i32 [[ABS]]
  58 ;
  59   %zext = zext i31 %x to i32
  60   %abs = call i32 @llvm.abs.i32(i32 %zext, i1 false)
  61   ret i32 %abs
  62 }
  63
  64 define <3 x i82> @lshr_abs(<3 x i82> %x) {
  65 ; CHECK-LABEL: @lshr_abs(
  66 ; CHECK-NEXT:    [[LSHR:%.*]] = lshr <3 x i82> [[X:%.*]], <i82 1, i82 1, i82 1>
  67 ; CHECK-NEXT:    [[ABS:%.*]] = call <3 x i82> @llvm.abs.v3i82(<3 x i82> [[LSHR]], i1 true)
  68 ; CHECK-NEXT:    ret <3 x i82> [[ABS]]
  69 ;
  70   %lshr = lshr <3 x i82> %x, <i82 1, i82 1, i82 1>
  71   %abs = call <3 x i82> @llvm.abs.v3i82(<3 x i82> %lshr, i1 true)
  72   ret <3 x i82> %abs
  73 }
  74
  75 define i32 @and_abs(i32 %x) {
  76 ; CHECK-LABEL: @and_abs(
  77 ; CHECK-NEXT:    [[AND:%.*]] = and i32 [[X:%.*]], 2147483644
  78 ; CHECK-NEXT:    [[ABS:%.*]] = call i32 @llvm.abs.i32(i32 [[AND]], i1 true)
  79 ; CHECK-NEXT:    ret i32 [[ABS]]
  80 ;
  81   %and = and i32 %x, 2147483644
  82   %abs = call i32 @llvm.abs.i32(i32 %and, i1 true)
  83   ret i32 %abs
  84 }
  85
  86 define <3 x i82> @select_abs(<3 x i1> %cond) {
  87 ; CHECK-LABEL: @select_abs(
  88 ; CHECK-NEXT:    [[SEL:%.*]] = select <3 x i1> [[COND:%.*]], <3 x i82> zeroinitializer, <3 x i82> <i82 2147483647, i82 42, i82 1>
  89 ; CHECK-NEXT:    [[ABS:%.*]] = call <3 x i82> @llvm.abs.v3i82(<3 x i82> [[SEL]], i1 false)
  90 ; CHECK-NEXT:    ret <3 x i82> [[ABS]]
  91 ;
  92   %sel = select <3 x i1> %cond, <3 x i82> zeroinitializer, <3 x i82> <i82 2147483647, i82 42, i82 1>
  93   %abs = call <3 x i82> @llvm.abs.v3i82(<3 x i82> %sel, i1 false)
  94   ret <3 x i82> %abs
  95 }
  96
  97 declare void @llvm.assume(i1)
  98
  99 define i32 @assume_abs(i32 %x) {
 100 ; CHECK-LABEL: @assume_abs(
 101 ; CHECK-NEXT:    [[ASSUME:%.*]] = icmp sge i32 [[X:%.*]], 0
 102 ; CHECK-NEXT:    call void @llvm.assume(i1 [[ASSUME]])
 103 ; CHECK-NEXT:    [[ABS:%.*]] = call i32 @llvm.abs.i32(i32 [[X]], i1 true)
 104 ; CHECK-NEXT:    ret i32 [[ABS]]
 105 ;
 106   %assume = icmp sge i32 %x, 0
 107   call void @llvm.assume(i1 %assume)
 108   %abs = call i32 @llvm.abs.i32(i32 %x, i1 true)
 109   ret i32 %abs
 110 }
 111
 112 define i1 @abs_nsw_must_be_positive(i32 %x) {
 113 ; CHECK-LABEL: @abs_nsw_must_be_positive(
 114 ; CHECK-NEXT:    ret i1 true
 115 ;
 116   %abs = call i32 @llvm.abs.i32(i32 %x, i1 true)
 117   %c2 = icmp sge i32 %abs, 0
 118   ret i1 %c2
 119 }
 120
 121 define <4 x i1> @abs_nsw_must_be_positive_vec(<4 x i32> %x) {
 122 ; CHECK-LABEL: @abs_nsw_must_be_positive_vec(
 123 ; CHECK-NEXT:    ret <4 x i1> <i1 true, i1 true, i1 true, i1 true>
 124 ;
 125   %abs = call <4 x i32> @llvm.abs.v4i32(<4 x i32> %x, i1 true)
 126   %c2 = icmp sge <4 x i32> %abs, zeroinitializer
 127   ret <4 x i1> %c2
 128 }
 129
 130 ; Negative test, no nsw provides no information about the sign bit of the result.
 131 define i1 @abs_nonsw(i32 %x) {
 132 ; CHECK-LABEL: @abs_nonsw(
 133 ; CHECK-NEXT:    [[ABS:%.*]] = call i32 @llvm.abs.i32(i32 [[X:%.*]], i1 false)
 134 ; CHECK-NEXT:    [[C2:%.*]] = icmp sge i32 [[ABS]], 0
 135 ; CHECK-NEXT:    ret i1 [[C2]]
 136 ;
 137   %abs = call i32 @llvm.abs.i32(i32 %x, i1 false)
 138   %c2 = icmp sge i32 %abs, 0
 139   ret i1 %c2
 140 }
 141
 142 define <4 x i1> @abs_nonsw_vec(<4 x i32> %x) {
 143 ; CHECK-LABEL: @abs_nonsw_vec(
 144 ; CHECK-NEXT:    [[ABS:%.*]] = call <4 x i32> @llvm.abs.v4i32(<4 x i32> [[X:%.*]], i1 false)
 145 ; CHECK-NEXT:    [[C2:%.*]] = icmp sge <4 x i32> [[ABS]], zeroinitializer
 146 ; CHECK-NEXT:    ret <4 x i1> [[C2]]
 147 ;
 148   %abs = call <4 x i32> @llvm.abs.v4i32(<4 x i32> %x, i1 false)
 149   %c2 = icmp sge <4 x i32> %abs, zeroinitializer
 150   ret <4 x i1> %c2
 151 }
 152
 153 define i1 @abs_known_positive_input_compare(i31 %x) {
 154 ; CHECK-LABEL: @abs_known_positive_input_compare(
 155 ; CHECK-NEXT:    ret i1 true
 156 ;
 157   %zext = zext i31 %x to i32
 158   %abs = call i32 @llvm.abs.i32(i32 %zext, i1 false)
 159   %c2 = icmp sge i32 %abs, 0
 160   ret i1 %c2
 161 }
 162
 163 define <4 x i1> @abs_known_positive_input_compare_vec(<4 x i31> %x) {
 164 ; CHECK-LABEL: @abs_known_positive_input_compare_vec(
 165 ; CHECK-NEXT:    ret <4 x i1> <i1 true, i1 true, i1 true, i1 true>
 166 ;
 167   %zext = zext <4 x i31> %x to <4 x i32>
 168   %abs = call <4 x i32> @llvm.abs.v4i32(<4 x i32> %zext, i1 false)
 169   %c2 = icmp sge <4 x i32> %abs, zeroinitializer
 170   ret <4 x i1> %c2
 171 }
 172
 173 define i1 @abs_known_not_int_min(i32 %x) {
 174 ; CHECK-LABEL: @abs_known_not_int_min(
 175 ; CHECK-NEXT:    ret i1 true
 176 ;
 177   %or = or i32 %x, 1
 178   %abs = call i32 @llvm.abs.i32(i32 %or, i1 false)
 179   %c2 = icmp sge i32 %abs, 0
 180   ret i1 %c2
 181 }
 182
 183 define <4 x i1> @abs_known_not_int_min_vec(<4 x i32> %x) {
 184 ; CHECK-LABEL: @abs_known_not_int_min_vec(
 185 ; CHECK-NEXT:    ret <4 x i1> <i1 true, i1 true, i1 true, i1 true>
 186 ;
 187   %or = or <4 x i32> %x, <i32 1, i32 1, i32 1, i32 1>
 188   %abs = call <4 x i32> @llvm.abs.v4i32(<4 x i32> %or, i1 false)
 189   %c2 = icmp sge <4 x i32> %abs, zeroinitializer
 190   ret <4 x i1> %c2
 191 }
 192
 193 ; If it's >= 0 it should be >= -1 as well.
 194 define i1 @abs_nsw_sge_neg(i32 %x) {
 195 ; CHECK-LABEL: @abs_nsw_sge_neg(
 196 ; CHECK-NEXT:    ret i1 true
 197 ;
 198   %abs = call i32 @llvm.abs.i32(i32 %x, i1 true)
 199   %c = icmp sge i32 %abs, -1
 200   ret i1 %c
 201 }
 202
 203 define i1 @abs_ule_int_min(i8 %x) {
 204 ; CHECK-LABEL: @abs_ule_int_min(
 205 ; CHECK-NEXT:    ret i1 true
 206 ;
 207   %abs = call i8 @llvm.abs.i8(i8 %x, i1 false)
 208   %c = icmp ule i8 %abs, 128
 209   ret i1 %c
 210 }
 211
 212 define i32 @select_abs_of_abs_eq(i32 %x) {
 213 ; CHECK-LABEL: @select_abs_of_abs_eq(
 214 ; CHECK-NEXT:    [[ABS:%.*]] = call i32 @llvm.abs.i32(i32 [[X:%.*]], i1 false)
 215 ; CHECK-NEXT:    ret i32 [[ABS]]
 216 ;
 217   %abs = call i32 @llvm.abs.i32(i32 %x, i1 false)
 218   %neg = sub i32 0, %abs
 219   %cmp = icmp eq i32 %x, 0
 220   %sel = select i1 %cmp, i32 %neg, i32 %abs
 221   ret i32 %sel
 222 }
 223
 224 ; The comparison is not with the same value we take abs() of, so this isn't the pattern.
 225 define i32 @select_abs_of_abs_eq_wrong(i32 %x, i32 %y) {
 226 ; CHECK-LABEL: @select_abs_of_abs_eq_wrong(
 227 ; CHECK-NEXT:    [[ABS:%.*]] = call i32 @llvm.abs.i32(i32 [[X:%.*]], i1 false)
 228 ; CHECK-NEXT:    [[NEG:%.*]] = sub i32 0, [[ABS]]
 229 ; CHECK-NEXT:    [[CMP:%.*]] = icmp eq i32 [[Y:%.*]], 0
 230 ; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[CMP]], i32 [[NEG]], i32 [[ABS]]
 231 ; CHECK-NEXT:    ret i32 [[SEL]]
 232 ;
 233   %abs = call i32 @llvm.abs.i32(i32 %x, i1 false)
 234   %neg = sub i32 0, %abs
 235   %cmp = icmp eq i32 %y, 0 ; not %x
 236   %sel = select i1 %cmp, i32 %neg, i32 %abs
 237   ret i32 %sel
 238 }
 239
 240 define i32 @select_abs_of_abs_ne(i32 %x) {
 241 ; CHECK-LABEL: @select_abs_of_abs_ne(
 242 ; CHECK-NEXT:    [[ABS:%.*]] = call i32 @llvm.abs.i32(i32 [[X:%.*]], i1 false)
 243 ; CHECK-NEXT:    ret i32 [[ABS]]
 244 ;
 245   %abs = call i32 @llvm.abs.i32(i32 %x, i1 false)
 246   %neg = sub i32 0, %abs
 247   %cmp = icmp ne i32 %x, 0
 248   %sel = select i1 %cmp, i32 %abs, i32 %neg
 249   ret i32 %sel
 250 }
 251
 252 define i32 @select_nabs_of_abs_eq(i32 %x) {
 253 ; CHECK-LABEL: @select_nabs_of_abs_eq(
 254 ; CHECK-NEXT:    [[ABS:%.*]] = call i32 @llvm.abs.i32(i32 [[X:%.*]], i1 false)
 255 ; CHECK-NEXT:    [[NEG:%.*]] = sub i32 0, [[ABS]]
 256 ; CHECK-NEXT:    ret i32 [[NEG]]
 257 ;
 258   %abs = call i32 @llvm.abs.i32(i32 %x, i1 false)
 259   %neg = sub i32 0, %abs
 260   %cmp = icmp eq i32 %x, 0
 261   %sel = select i1 %cmp, i32 %abs, i32 %neg
 262   ret i32 %sel
 263 }
 264
 265 ; The comparison is not with the same value we take abs() of, so this isn't the pattern.
 266 define i32 @select_nabs_of_abs_eq_wrong(i32 %x, i32 %y) {
 267 ; CHECK-LABEL: @select_nabs_of_abs_eq_wrong(
 268 ; CHECK-NEXT:    [[ABS:%.*]] = call i32 @llvm.abs.i32(i32 [[X:%.*]], i1 false)
 269 ; CHECK-NEXT:    [[NEG:%.*]] = sub i32 0, [[ABS]]
 270 ; CHECK-NEXT:    [[CMP:%.*]] = icmp eq i32 [[Y:%.*]], 0
 271 ; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[CMP]], i32 [[ABS]], i32 [[NEG]]
 272 ; CHECK-NEXT:    ret i32 [[SEL]]
 273 ;
 274   %abs = call i32 @llvm.abs.i32(i32 %x, i1 false)
 275   %neg = sub i32 0, %abs
 276   %cmp = icmp eq i32 %y, 0
 277   %sel = select i1 %cmp, i32 %abs, i32 %neg
 278   ret i32 %sel
 279 }
 280
 281 define i32 @select_nabs_of_abs_ne(i32 %x) {
 282 ; CHECK-LABEL: @select_nabs_of_abs_ne(
 283 ; CHECK-NEXT:    [[ABS:%.*]] = call i32 @llvm.abs.i32(i32 [[X:%.*]], i1 false)
 284 ; CHECK-NEXT:    [[NEG:%.*]] = sub i32 0, [[ABS]]
 285 ; CHECK-NEXT:    ret i32 [[NEG]]
 286 ;
 287   %abs = call i32 @llvm.abs.i32(i32 %x, i1 false)
 288   %neg = sub i32 0, %abs
 289   %cmp = icmp ne i32 %x, 0
 290   %sel = select i1 %cmp, i32 %neg, i32 %abs
 291   ret i32 %sel
 292 }