test/Transforms/InstCombine/memchr.ll

   1 ; Test that the memchr library call simplifier works correctly.
   2 ; RUN: opt < %s -instcombine -S | FileCheck %s
   3
   4 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"
   5
   6 @hello = constant [14 x i8] c"hello world\5Cn\00"
   7 @hellonull = constant [14 x i8] c"hello\00world\5Cn\00"
   8 @null = constant [1 x i8] zeroinitializer
   9 @newlines = constant [3 x i8] c"\0D\0A\00"
  10 @single = constant [2 x i8] c"\1F\00"
  11 @spaces = constant [4 x i8] c" \0D\0A\00"
  12 @negative = constant [3 x i8] c"\FF\FE\00"
  13 @chp = global i8* zeroinitializer
  14
  15 declare i8* @memchr(i8*, i32, i32)
  16
  17 define void @test1() {
  18 ; CHECK-LABEL: @test1
  19 ; CHECK: store i8* getelementptr inbounds ([14 x i8], [14 x i8]* @hello, i32 0, i32 6)
  20 ; CHECK-NOT: call i8* @memchr
  21 ; CHECK: 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: store i8* null, i8** @chp, align 4
  32 ; CHECK-NOT: call i8* @memchr
  33 ; CHECK: ret void
  34
  35   %str = getelementptr [1 x i8], [1 x i8]* @null, i32 0, i32 0
  36   %dst = call i8* @memchr(i8* %str, i32 119, i32 1)
  37   store i8* %dst, i8** @chp
  38   ret void
  39 }
  40
  41 define void @test3() {
  42 ; CHECK-LABEL: @test3
  43 ; CHECK: store i8* getelementptr inbounds ([14 x i8], [14 x i8]* @hello, i32 0, i32 13)
  44 ; CHECK-NOT: call i8* @memchr
  45 ; CHECK: ret void
  46
  47   %src = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
  48   %dst = call i8* @memchr(i8* %src, i32 0, i32 14)
  49   store i8* %dst, i8** @chp
  50   ret void
  51 }
  52
  53 define void @test4(i32 %chr) {
  54 ; CHECK-LABEL: @test4
  55 ; CHECK: call i8* @memchr
  56 ; CHECK-NOT: call i8* @memchr
  57 ; CHECK: ret void
  58
  59   %src = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
  60   %dst = call i8* @memchr(i8* %src, i32 %chr, i32 14)
  61   store i8* %dst, i8** @chp
  62   ret void
  63 }
  64
  65 define void @test5() {
  66 ; CHECK-LABEL: @test5
  67 ; CHECK: store i8* getelementptr inbounds ([14 x i8], [14 x i8]* @hello, i32 0, i32 13)
  68 ; CHECK-NOT: call i8* @memchr
  69 ; CHECK: ret void
  70
  71   %src = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
  72   %dst = call i8* @memchr(i8* %src, i32 65280, i32 14)
  73   store i8* %dst, i8** @chp
  74   ret void
  75 }
  76
  77 define void @test6() {
  78 ; CHECK-LABEL: @test6
  79 ; CHECK: store i8* getelementptr inbounds ([14 x i8], [14 x i8]* @hello, i32 0, i32 6)
  80 ; CHECK-NOT: call i8* @memchr
  81 ; CHECK: ret void
  82
  83   %src = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
  84 ; Overflow, but we still find the right thing.
  85   %dst = call i8* @memchr(i8* %src, i32 119, i32 100)
  86   store i8* %dst, i8** @chp
  87   ret void
  88 }
  89
  90 define void @test7() {
  91 ; CHECK-LABEL: @test7
  92 ; CHECK: store i8* null, i8** @chp, align 4
  93 ; CHECK-NOT: call i8* @memchr
  94 ; CHECK: ret void
  95
  96   %src = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
  97 ; Overflow
  98   %dst = call i8* @memchr(i8* %src, i32 120, i32 100)
  99   store i8* %dst, i8** @chp
 100   ret void
 101 }
 102
 103 define void @test8() {
 104 ; CHECK-LABEL: @test8
 105 ; CHECK: store i8* getelementptr inbounds ([14 x i8], [14 x i8]* @hellonull, i32 0, i32 6)
 106 ; CHECK-NOT: call i8* @memchr
 107 ; CHECK: ret void
 108
 109   %str = getelementptr [14 x i8], [14 x i8]* @hellonull, i32 0, i32 0
 110   %dst = call i8* @memchr(i8* %str, i32 119, i32 14)
 111   store i8* %dst, i8** @chp
 112   ret void
 113 }
 114
 115 define void @test9() {
 116 ; CHECK-LABEL: @test9
 117 ; CHECK: store i8* getelementptr inbounds ([14 x i8], [14 x i8]* @hellonull, i32 0, i32 6)
 118 ; CHECK-NOT: call i8* @memchr
 119 ; CHECK: ret void
 120
 121   %str = getelementptr [14 x i8], [14 x i8]* @hellonull, i32 0, i32 2
 122   %dst = call i8* @memchr(i8* %str, i32 119, i32 12)
 123   store i8* %dst, i8** @chp
 124   ret void
 125 }
 126
 127 define void @test10() {
 128 ; CHECK-LABEL: @test10
 129 ; CHECK: store i8* null, i8** @chp, align 4
 130 ; CHECK-NOT: call i8* @memchr
 131 ; CHECK: ret void
 132
 133   %str = getelementptr [14 x i8], [14 x i8]* @hello, i32 0, i32 0
 134   %dst = call i8* @memchr(i8* %str, i32 119, i32 6)
 135   store i8* %dst, i8** @chp
 136   ret void
 137 }
 138
 139 ; Check transformation memchr("\r\n", C, 2) != nullptr -> (C & 9216) != 0
 140 define i1 @test11(i32 %C) {
 141 ; CHECK-LABEL: @test11
 142 ; CHECK-NEXT: [[TRUNC:%.*]] = trunc i32 %C to i16
 143 ; CHECK-NEXT: %memchr.bounds = icmp ult i16 [[TRUNC]], 16
 144 ; CHECK-NEXT: [[SHL:%.*]] = shl i16 1, [[TRUNC]]
 145 ; CHECK-NEXT: [[AND:%.*]] = and i16 [[SHL]], 9216
 146 ; CHECK-NEXT: %memchr.bits = icmp ne i16 [[AND]], 0
 147 ; CHECK-NEXT: %memchr = and i1 %memchr.bounds, %memchr.bits
 148 ; CHECK-NEXT: ret i1 %memchr
 149
 150   %dst = call i8* @memchr(i8* getelementptr inbounds ([3 x i8], [3 x i8]* @newlines, i64 0, i64 0), i32 %C, i32 2)
 151   %cmp = icmp ne i8* %dst, null
 152   ret i1 %cmp
 153 }
 154
 155 ; No 64 bits here
 156 define i1 @test12(i32 %C) {
 157 ; CHECK-LABEL: @test12
 158 ; CHECK-NEXT: %dst = call i8* @memchr(i8* getelementptr inbounds ([4 x i8], [4 x i8]* @spaces, i32 0, i32 0), i32 %C, i32 3)
 159 ; CHECK-NEXT: %cmp = icmp ne i8* %dst, null
 160 ; CHECK-NEXT: ret i1 %cmp
 161
 162   %dst = call i8* @memchr(i8* getelementptr inbounds ([4 x i8], [4 x i8]* @spaces, i64 0, i64 0), i32 %C, i32 3)
 163   %cmp = icmp ne i8* %dst, null
 164   ret i1 %cmp
 165 }
 166
 167 define i1 @test13(i32 %C) {
 168 ; CHECK-LABEL: @test13
 169 ; CHECK-NEXT: %memchr.bounds = icmp ult i32 %C, 32
 170 ; CHECK-NEXT: [[SHL:%.*]] = shl i32 1, %C
 171 ; CHECK-NEXT: [[AND:%.*]] = and i32 [[SHL]], -2147483647
 172 ; CHECK-NEXT: %memchr.bits = icmp ne i32 [[AND]], 0
 173 ; CHECK-NEXT: %memchr = and i1 %memchr.bounds, %memchr.bits
 174 ; CHECK-NEXT: ret i1 %memchr
 175
 176   %dst = call i8* @memchr(i8* getelementptr inbounds ([2 x i8], [2 x i8]* @single, i64 0, i64 0), i32 %C, i32 2)
 177   %cmp = icmp ne i8* %dst, null
 178   ret i1 %cmp
 179 }
 180
 181 define i1 @test14(i32 %C) {
 182 ; CHECK-LABEL: @test14
 183 ; CHECK-NEXT: icmp eq i32 %C, 31
 184 ; CHECK-NEXT: ret
 185
 186   %dst = call i8* @memchr(i8* getelementptr inbounds ([2 x i8], [2 x i8]* @single, i64 0, i64 0), i32 %C, i32 1)
 187   %cmp = icmp ne i8* %dst, null
 188   ret i1 %cmp
 189 }
 190
 191 define i1 @test15(i32 %C) {
 192 ; CHECK-LABEL: @test15
 193 ; CHECK-NEXT: %dst = call i8* @memchr(i8* getelementptr inbounds ([3 x i8], [3 x i8]* @negative, i32 0, i32 0), i32 %C, i32 3)
 194 ; CHECK-NEXT: %cmp = icmp ne i8* %dst, null
 195 ; CHECK-NEXT: ret i1 %cmp
 196
 197   %dst = call i8* @memchr(i8* getelementptr inbounds ([3 x i8], [3 x i8]* @negative, i64 0, i64 0), i32 %C, i32 3)
 198   %cmp = icmp ne i8* %dst, null
 199   ret i1 %cmp
 200 }