234 lines · plain
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py2; Test that the memchr library call simplifier works correctly.3; RUN: opt < %s -passes=instcombine -S | FileCheck %s4 5target 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] zeroinitializer10@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 ptr zeroinitializer15 16declare ptr @memchr(ptr, i32, i32)17 18define void @test1() {19; CHECK-LABEL: @test1(20; CHECK-NEXT: store ptr getelementptr inbounds nuw (i8, ptr @hello, i32 6), ptr @chp, align 421; CHECK-NEXT: ret void22;23 %dst = call ptr @memchr(ptr @hello, i32 119, i32 14)24 store ptr %dst, ptr @chp25 ret void26}27 28define void @test2() {29; CHECK-LABEL: @test2(30; CHECK-NEXT: store ptr null, ptr @chp, align 431; CHECK-NEXT: ret void32;33 %dst = call ptr @memchr(ptr @null, i32 119, i32 1)34 store ptr %dst, ptr @chp35 ret void36}37 38define void @test3() {39; CHECK-LABEL: @test3(40; CHECK-NEXT: store ptr getelementptr inbounds nuw (i8, ptr @hello, i32 13), ptr @chp, align 441; CHECK-NEXT: ret void42;43 %dst = call ptr @memchr(ptr @hello, i32 0, i32 14)44 store ptr %dst, ptr @chp45 ret void46}47 48define void @test4(i32 %chr) {49; CHECK-LABEL: @test4(50; CHECK-NEXT: [[DST:%.*]] = call ptr @memchr(ptr noundef nonnull dereferenceable(1) @hello, i32 [[CHR:%.*]], i32 14)51; CHECK-NEXT: store ptr [[DST]], ptr @chp, align 452; CHECK-NEXT: ret void53;54 %dst = call ptr @memchr(ptr @hello, i32 %chr, i32 14)55 store ptr %dst, ptr @chp56 ret void57}58 59define void @test5() {60; CHECK-LABEL: @test5(61; CHECK-NEXT: store ptr getelementptr inbounds nuw (i8, ptr @hello, i32 13), ptr @chp, align 462; CHECK-NEXT: ret void63;64 %dst = call ptr @memchr(ptr @hello, i32 65280, i32 14)65 store ptr %dst, ptr @chp66 ret void67}68 69define void @test6() {70; CHECK-LABEL: @test6(71; CHECK-NEXT: store ptr getelementptr inbounds nuw (i8, ptr @hello, i32 6), ptr @chp, align 472; CHECK-NEXT: ret void73;74; Overflow, but we still find the right thing.75 %dst = call ptr @memchr(ptr @hello, i32 119, i32 100)76 store ptr %dst, ptr @chp77 ret void78}79 80define void @test7() {81; CHECK-LABEL: @test7(82; CHECK-NEXT: store ptr null, ptr @chp, align 483; CHECK-NEXT: ret void84;85; Overflow86 %dst = call ptr @memchr(ptr @hello, i32 120, i32 100)87 store ptr %dst, ptr @chp88 ret void89}90 91define void @test8() {92; CHECK-LABEL: @test8(93; CHECK-NEXT: store ptr getelementptr inbounds nuw (i8, ptr @hellonull, i32 6), ptr @chp, align 494; CHECK-NEXT: ret void95;96 %dst = call ptr @memchr(ptr @hellonull, i32 119, i32 14)97 store ptr %dst, ptr @chp98 ret void99}100 101define void @test9() {102; CHECK-LABEL: @test9(103; CHECK-NEXT: store ptr getelementptr inbounds nuw (i8, ptr @hellonull, i32 6), ptr @chp, align 4104; CHECK-NEXT: ret void105;106 %str = getelementptr [14 x i8], ptr @hellonull, i32 0, i32 2107 %dst = call ptr @memchr(ptr %str, i32 119, i32 12)108 store ptr %dst, ptr @chp109 ret void110}111 112define void @test10() {113; CHECK-LABEL: @test10(114; CHECK-NEXT: store ptr null, ptr @chp, align 4115; CHECK-NEXT: ret void116;117 %dst = call ptr @memchr(ptr @hello, i32 119, i32 6)118 store ptr %dst, ptr @chp119 ret void120}121 122; Check transformation memchr("\r\n", C, 3) != nullptr -> (C & 9217) != 0123define i1 @test11(i32 %C) {124; CHECK-LABEL: @test11(125; CHECK-NEXT: [[TMP1:%.*]] = trunc i32 [[C:%.*]] to i16126; CHECK-NEXT: [[TMP2:%.*]] = and i16 [[TMP1]], 255127; CHECK-NEXT: [[MEMCHR_BOUNDS:%.*]] = icmp samesign ult i16 [[TMP2]], 16128; CHECK-NEXT: [[TMP3:%.*]] = shl nuw i16 1, [[TMP2]]129; CHECK-NEXT: [[TMP4:%.*]] = and i16 [[TMP3]], 9217130; CHECK-NEXT: [[MEMCHR_BITS:%.*]] = icmp ne i16 [[TMP4]], 0131; CHECK-NEXT: [[MEMCHR:%.*]] = select i1 [[MEMCHR_BOUNDS]], i1 [[MEMCHR_BITS]], i1 false132; CHECK-NEXT: ret i1 [[MEMCHR]]133;134 %dst = call ptr @memchr(ptr @newlines, i32 %C, i32 3)135 %cmp = icmp ne ptr %dst, null136 ret i1 %cmp137}138 139; No 64 bits here140define i1 @test12(i32 %C) {141; CHECK-LABEL: @test12(142; CHECK-NEXT: [[DST:%.*]] = call ptr @memchr(ptr noundef nonnull dereferenceable(1) @spaces, i32 [[C:%.*]], i32 3)143; CHECK-NEXT: [[CMP:%.*]] = icmp ne ptr [[DST]], null144; CHECK-NEXT: ret i1 [[CMP]]145;146 %dst = call ptr @memchr(ptr @spaces, i32 %C, i32 3)147 %cmp = icmp ne ptr %dst, null148 ret i1 %cmp149}150 151define i1 @test13(i32 %C) {152; CHECK-LABEL: @test13(153; CHECK-NEXT: [[TMP1:%.*]] = trunc i32 [[C:%.*]] to i8154; CHECK-NEXT: [[TMP2:%.*]] = icmp eq i8 [[TMP1]], 0155; CHECK-NEXT: [[TMP3:%.*]] = icmp eq i8 [[TMP1]], 31156; CHECK-NEXT: [[CMP:%.*]] = or i1 [[TMP3]], [[TMP2]]157; CHECK-NEXT: ret i1 [[CMP]]158;159 %dst = call ptr @memchr(ptr @single, i32 %C, i32 2)160 %cmp = icmp ne ptr %dst, null161 ret i1 %cmp162}163 164define i1 @test14(i32 %C) {165; CHECK-LABEL: @test14(166; CHECK-NEXT: [[TMP1:%.*]] = and i32 [[C:%.*]], 255167; CHECK-NEXT: [[MEMCHR_CHAR0CMP:%.*]] = icmp eq i32 [[TMP1]], 31168; CHECK-NEXT: ret i1 [[MEMCHR_CHAR0CMP]]169;170 %dst = call ptr @memchr(ptr @single, i32 %C, i32 1)171 %cmp = icmp ne ptr %dst, null172 ret i1 %cmp173}174 175define i1 @test15(i32 %C, i32 %n) {176; CHECK-LABEL: @test15(177; CHECK-NEXT: [[DST:%.*]] = call ptr @memchr(ptr nonnull @negative, i32 [[C:%.*]], i32 [[N:%.*]])178; CHECK-NEXT: [[CMP:%.*]] = icmp ne ptr [[DST]], null179; CHECK-NEXT: ret i1 [[CMP]]180;181 %dst = call ptr @memchr(ptr @negative, i32 %C, i32 %n)182 %cmp = icmp ne ptr %dst, null183 ret i1 %cmp184}185 186@s = internal constant [1 x i8] [i8 0], align 1187define ptr @pr32124() {188; CHECK-LABEL: @pr32124(189; CHECK-NEXT: ret ptr @s190;191 %res = tail call ptr @memchr(ptr @s, i32 0, i32 1)192 ret ptr %res193}194 195define ptr @test16(ptr %str, i32 %c, i32 %n) {196; CHECK-LABEL: @test16(197; CHECK-NEXT: [[RET:%.*]] = call ptr @memchr(ptr [[STR:%.*]], i32 [[C:%.*]], i32 [[N:%.*]])198; CHECK-NEXT: ret ptr [[RET]]199;200 201 %ret = call ptr @memchr(ptr %str, i32 %c, i32 %n)202 ret ptr %ret203}204 205define ptr @test17(ptr %str, i32 %c, i32 %n) {206; CHECK-LABEL: @test17(207; CHECK-NEXT: [[RET:%.*]] = call ptr @memchr(ptr nonnull [[STR:%.*]], i32 [[C:%.*]], i32 [[N:%.*]])208; CHECK-NEXT: ret ptr [[RET]]209;210 211 %ret = call ptr @memchr(ptr nonnull %str, i32 %c, i32 %n)212 ret ptr %ret213}214 215define ptr @test18(ptr %str, i32 %c) {216; CHECK-LABEL: @test18(217; CHECK-NEXT: [[RET:%.*]] = call ptr @memchr(ptr noundef nonnull dereferenceable(1) [[STR:%.*]], i32 [[C:%.*]], i32 5)218; CHECK-NEXT: ret ptr [[RET]]219;220 221 %ret = call ptr @memchr(ptr %str, i32 %c, i32 5)222 ret ptr %ret223}224 225define ptr @test19(ptr %str, i32 %c) null_pointer_is_valid {226; CHECK-LABEL: @test19(227; CHECK-NEXT: [[RET:%.*]] = call ptr @memchr(ptr noundef [[STR:%.*]], i32 [[C:%.*]], i32 5)228; CHECK-NEXT: ret ptr [[RET]]229;230 231 %ret = call ptr @memchr(ptr %str, i32 %c, i32 5)232 ret ptr %ret233}234