llvm/test/Transforms/InstCombine/memchr.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; Test that the memchr library call simplifier works correctly.
   3 ; RUN: opt < %s -instcombine -S | FileCheck %s
   4
   5 target datalayout = "e-p:32:32:32-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:32:64-f32:32:32-f64:32:64-v64:64:64-v128:128:128-a0:0:64-f80:128:128-n8:16:32"
   6
   7 @hello = constant [14 x i8] c"hello world\5Cn\00"
   8 @hellonull = constant [14 x i8] c"hello\00world\5Cn\00"
   9 @null = constant [1 x i8] zeroinitializer
  10 @newlines = constant [3 x i8] c"\0D\0A\00"
  11 @single = constant [2 x i8] c"\1F\00"
  12 @spaces = constant [4 x i8] c" \0D\0A\00"
  13 @negative = constant [3 x i8] c"\FF\FE\00"
  14 @chp = global i8* zeroinitializer
  15
  16 declare i8* @memchr(i8*, i32, i32)
  17
  18 define void @test1() {
  19 ; CHECK-LABEL: @test1(
  20 ; CHECK-NEXT:    store i8* getelementptr inbounds ([14 x i8], [14 x i8]* @hello, i32 0, i32 6), i8** @chp, align 4
  21 ; CHECK-NEXT:    ret void
  22 ;
  23   %str = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
  24   %dst = call i8* @memchr(i8* %str, i32 119, i32 14)
  25   store i8* %dst, i8** @chp
  26   ret void
  27 }
  28
  29 define void @test2() {
  30 ; CHECK-LABEL: @test2(
  31 ; CHECK-NEXT:    store i8* null, i8** @chp, align 4
  32 ; CHECK-NEXT:    ret void
  33 ;
  34   %str = getelementptr [1 x i8], [1 x i8]* @null, i32 0, i32 0
  35   %dst = call i8* @memchr(i8* %str, i32 119, i32 1)
  36   store i8* %dst, i8** @chp
  37   ret void
  38 }
  39
  40 define void @test3() {
  41 ; CHECK-LABEL: @test3(
  42 ; CHECK-NEXT:    store i8* getelementptr inbounds ([14 x i8], [14 x i8]* @hello, i32 0, i32 13), i8** @chp, align 4
  43 ; CHECK-NEXT:    ret void
  44 ;
  45   %src = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
  46   %dst = call i8* @memchr(i8* %src, i32 0, i32 14)
  47   store i8* %dst, i8** @chp
  48   ret void
  49 }
  50
  51 define void @test4(i32 %chr) {
  52 ; CHECK-LABEL: @test4(
  53 ; CHECK-NEXT:    [[DST:%.*]] = call i8* @memchr(i8* noundef nonnull dereferenceable(14) getelementptr inbounds ([14 x i8], [14 x i8]* @hello, i32 0, i32 0), i32 [[CHR:%.*]], i32 14)
  54 ; CHECK-NEXT:    store i8* [[DST]], i8** @chp, align 4
  55 ; CHECK-NEXT:    ret void
  56 ;
  57   %src = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
  58   %dst = call i8* @memchr(i8* %src, i32 %chr, i32 14)
  59   store i8* %dst, i8** @chp
  60   ret void
  61 }
  62
  63 define void @test5() {
  64 ; CHECK-LABEL: @test5(
  65 ; CHECK-NEXT:    store i8* getelementptr inbounds ([14 x i8], [14 x i8]* @hello, i32 0, i32 13), i8** @chp, align 4
  66 ; CHECK-NEXT:    ret void
  67 ;
  68   %src = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
  69   %dst = call i8* @memchr(i8* %src, i32 65280, i32 14)
  70   store i8* %dst, i8** @chp
  71   ret void
  72 }
  73
  74 define void @test6() {
  75 ; CHECK-LABEL: @test6(
  76 ; CHECK-NEXT:    store i8* getelementptr inbounds ([14 x i8], [14 x i8]* @hello, i32 0, i32 6), i8** @chp, align 4
  77 ; CHECK-NEXT:    ret void
  78 ;
  79   %src = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
  80 ; Overflow, but we still find the right thing.
  81   %dst = call i8* @memchr(i8* %src, i32 119, i32 100)
  82   store i8* %dst, i8** @chp
  83   ret void
  84 }
  85
  86 define void @test7() {
  87 ; CHECK-LABEL: @test7(
  88 ; CHECK-NEXT:    store i8* null, i8** @chp, align 4
  89 ; CHECK-NEXT:    ret void
  90 ;
  91   %src = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
  92 ; Overflow
  93   %dst = call i8* @memchr(i8* %src, i32 120, i32 100)
  94   store i8* %dst, i8** @chp
  95   ret void
  96 }
  97
  98 define void @test8() {
  99 ; CHECK-LABEL: @test8(
 100 ; CHECK-NEXT:    store i8* getelementptr inbounds ([14 x i8], [14 x i8]* @hellonull, i32 0, i32 6), i8** @chp, align 4
 101 ; CHECK-NEXT:    ret void
 102 ;
 103   %str = getelementptr [14 x i8], [14 x i8]* @hellonull, i32 0, i32 0
 104   %dst = call i8* @memchr(i8* %str, i32 119, i32 14)
 105   store i8* %dst, i8** @chp
 106   ret void
 107 }
 108
 109 define void @test9() {
 110 ; CHECK-LABEL: @test9(
 111 ; CHECK-NEXT:    store i8* getelementptr inbounds ([14 x i8], [14 x i8]* @hellonull, i32 0, i32 6), i8** @chp, align 4
 112 ; CHECK-NEXT:    ret void
 113 ;
 114   %str = getelementptr [14 x i8], [14 x i8]* @hellonull, i32 0, i32 2
 115   %dst = call i8* @memchr(i8* %str, i32 119, i32 12)
 116   store i8* %dst, i8** @chp
 117   ret void
 118 }
 119
 120 define void @test10() {
 121 ; CHECK-LABEL: @test10(
 122 ; CHECK-NEXT:    store i8* null, i8** @chp, align 4
 123 ; CHECK-NEXT:    ret void
 124 ;
 125   %str = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
 126   %dst = call i8* @memchr(i8* %str, i32 119, i32 6)
 127   store i8* %dst, i8** @chp
 128   ret void
 129 }
 130
 131 ; Check transformation memchr("\r\n", C, 2) != nullptr -> (C & 9216) != 0
 132 define i1 @test11(i32 %C) {
 133 ; CHECK-LABEL: @test11(
 134 ; CHECK-NEXT:    [[TMP1:%.*]] = trunc i32 [[C:%.*]] to i16
 135 ; CHECK-NEXT:    [[TMP2:%.*]] = and i16 [[TMP1]], 255
 136 ; CHECK-NEXT:    [[MEMCHR_BOUNDS:%.*]] = icmp ult i16 [[TMP2]], 16
 137 ; CHECK-NEXT:    [[TMP3:%.*]] = shl i16 1, [[TMP2]]
 138 ; CHECK-NEXT:    [[TMP4:%.*]] = and i16 [[TMP3]], 9216
 139 ; CHECK-NEXT:    [[MEMCHR_BITS:%.*]] = icmp ne i16 [[TMP4]], 0
 140 ; CHECK-NEXT:    [[MEMCHR:%.*]] = select i1 [[MEMCHR_BOUNDS]], i1 [[MEMCHR_BITS]], i1 false
 141 ; CHECK-NEXT:    ret i1 [[MEMCHR]]
 142 ;
 143   %dst = call i8* @memchr(i8* getelementptr inbounds ([3 x i8], [3 x i8]* @newlines, i64 0, i64 0), i32 %C, i32 2)
 144   %cmp = icmp ne i8* %dst, null
 145   ret i1 %cmp
 146 }
 147
 148 ; No 64 bits here
 149 define i1 @test12(i32 %C) {
 150 ; CHECK-LABEL: @test12(
 151 ; CHECK-NEXT:    [[DST:%.*]] = call i8* @memchr(i8* noundef nonnull dereferenceable(3) getelementptr inbounds ([4 x i8], [4 x i8]* @spaces, i32 0, i32 0), i32 [[C:%.*]], i32 3)
 152 ; CHECK-NEXT:    [[CMP:%.*]] = icmp ne i8* [[DST]], null
 153 ; CHECK-NEXT:    ret i1 [[CMP]]
 154 ;
 155   %dst = call i8* @memchr(i8* getelementptr inbounds ([4 x i8], [4 x i8]* @spaces, i64 0, i64 0), i32 %C, i32 3)
 156   %cmp = icmp ne i8* %dst, null
 157   ret i1 %cmp
 158 }
 159
 160 define i1 @test13(i32 %C) {
 161 ; CHECK-LABEL: @test13(
 162 ; CHECK-NEXT:    [[TMP1:%.*]] = and i32 [[C:%.*]], 255
 163 ; CHECK-NEXT:    [[MEMCHR_BOUNDS:%.*]] = icmp ult i32 [[TMP1]], 32
 164 ; CHECK-NEXT:    [[TMP2:%.*]] = shl i32 1, [[TMP1]]
 165 ; CHECK-NEXT:    [[TMP3:%.*]] = and i32 [[TMP2]], -2147483647
 166 ; CHECK-NEXT:    [[MEMCHR_BITS:%.*]] = icmp ne i32 [[TMP3]], 0
 167 ; CHECK-NEXT:    [[MEMCHR:%.*]] = select i1 [[MEMCHR_BOUNDS]], i1 [[MEMCHR_BITS]], i1 false
 168 ; CHECK-NEXT:    ret i1 [[MEMCHR]]
 169 ;
 170   %dst = call i8* @memchr(i8* getelementptr inbounds ([2 x i8], [2 x i8]* @single, i64 0, i64 0), i32 %C, i32 2)
 171   %cmp = icmp ne i8* %dst, null
 172   ret i1 %cmp
 173 }
 174
 175 define i1 @test14(i32 %C) {
 176 ; CHECK-LABEL: @test14(
 177 ; CHECK-NEXT:    [[TMP1:%.*]] = and i32 [[C:%.*]], 255
 178 ; CHECK-NEXT:    [[MEMCHR_BITS:%.*]] = icmp eq i32 [[TMP1]], 31
 179 ; CHECK-NEXT:    ret i1 [[MEMCHR_BITS]]
 180 ;
 181   %dst = call i8* @memchr(i8* getelementptr inbounds ([2 x i8], [2 x i8]* @single, i64 0, i64 0), i32 %C, i32 1)
 182   %cmp = icmp ne i8* %dst, null
 183   ret i1 %cmp
 184 }
 185
 186 define i1 @test15(i32 %C) {
 187 ; CHECK-LABEL: @test15(
 188 ; CHECK-NEXT:    [[DST:%.*]] = call i8* @memchr(i8* noundef nonnull dereferenceable(3) getelementptr inbounds ([3 x i8], [3 x i8]* @negative, i32 0, i32 0), i32 [[C:%.*]], i32 3)
 189 ; CHECK-NEXT:    [[CMP:%.*]] = icmp ne i8* [[DST]], null
 190 ; CHECK-NEXT:    ret i1 [[CMP]]
 191 ;
 192   %dst = call i8* @memchr(i8* getelementptr inbounds ([3 x i8], [3 x i8]* @negative, i64 0, i64 0), i32 %C, i32 3)
 193   %cmp = icmp ne i8* %dst, null
 194   ret i1 %cmp
 195 }
 196
 197 @s = internal constant [1 x i8] [i8 0], align 1
 198 define i8* @pr32124() {
 199 ; CHECK-LABEL: @pr32124(
 200 ; CHECK-NEXT:    ret i8* getelementptr inbounds ([1 x i8], [1 x i8]* @s, i32 0, i32 0)
 201 ;
 202   %res = tail call i8* @memchr(i8* getelementptr ([1 x i8], [1 x i8]* @s, i64 0, i64 0), i32 0, i32 1)
 203   ret i8* %res
 204 }
 205
 206 define i8* @test16(i8* %str, i32 %c, i32 %n) {
 207 ; CHECK-LABEL: @test16(
 208 ; CHECK-NEXT:    [[RET:%.*]] = call i8* @memchr(i8* [[STR:%.*]], i32 [[C:%.*]], i32 [[N:%.*]])
 209 ; CHECK-NEXT:    ret i8* [[RET]]
 210 ;
 211
 212   %ret = call i8* @memchr(i8* %str, i32 %c, i32 %n)
 213   ret i8* %ret
 214 }
 215
 216 define i8* @test17(i8* %str, i32 %c, i32 %n) {
 217 ; CHECK-LABEL: @test17(
 218 ; CHECK-NEXT:    [[RET:%.*]] = call i8* @memchr(i8* nonnull [[STR:%.*]], i32 [[C:%.*]], i32 [[N:%.*]])
 219 ; CHECK-NEXT:    ret i8* [[RET]]
 220 ;
 221
 222   %ret = call i8* @memchr(i8* nonnull %str, i32 %c, i32 %n)
 223   ret i8* %ret
 224 }
 225
 226 define i8* @test18(i8* %str, i32 %c) {
 227 ; CHECK-LABEL: @test18(
 228 ; CHECK-NEXT:    [[RET:%.*]] = call i8* @memchr(i8* noundef nonnull dereferenceable(5) [[STR:%.*]], i32 [[C:%.*]], i32 5)
 229 ; CHECK-NEXT:    ret i8* [[RET]]
 230 ;
 231
 232   %ret = call i8* @memchr(i8* %str, i32 %c, i32 5)
 233   ret i8* %ret
 234 }
 235
 236 define i8* @test19(i8* %str, i32 %c) null_pointer_is_valid {
 237 ; CHECK-LABEL: @test19(
 238 ; CHECK-NEXT:    [[RET:%.*]] = call i8* @memchr(i8* noundef dereferenceable(5) [[STR:%.*]], i32 [[C:%.*]], i32 5)
 239 ; CHECK-NEXT:    ret i8* [[RET]]
 240 ;
 241
 242   %ret = call i8* @memchr(i8* %str, i32 %c, i32 5)
 243   ret i8* %ret
 244 }