Doomhowl-Interactive/llvm-project
0
0
Fork 0
Code Issues Pull Requests Actions 5 Packages Projects Releases Wiki Activity
Files
main
llvm-project/llvm/test/Transforms/AggressiveInstCombine/memchr.ll
Nikita Popov 2fe39f24b2 [AggressiveInstCombine] Avoid implicit truncation 
Cast char to unsigned char to match the unsigned ConstantInt
constructor.
2026-01-14 10:44:47 +01:00

213 lines
8.2 KiB
LLVM
Raw Permalink Blame History

; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
; RUN: opt -S -passes=aggressive-instcombine --memchr-inline-threshold=5 < %s | FileCheck %s
@str = constant [5 x i8] c"01\002\00", align 1
@str_long = constant [8 x i8] c"0123456\00", align 1
@str_hi = constant [3 x i8] c"\80\81\00"
declare ptr @memchr(ptr, i32, i64)
define i1 @test_memchr_null(i32 %x) !prof !0 {
; CHECK-LABEL: define i1 @test_memchr_null(
; CHECK-SAME: i32 [[X:%.*]]) !prof [[PROF0:![0-9]+]] {
; CHECK-NEXT: [[ENTRY:.*]]:
; CHECK-NEXT: [[TMP0:%.*]] = trunc i32 [[X]] to i8
; CHECK-NEXT: switch i8 [[TMP0]], label %[[ENTRY_SPLIT:.*]] [
; CHECK-NEXT: i8 48, label %[[MEMCHR_CASE:.*]]
; CHECK-NEXT: i8 49, label %[[MEMCHR_CASE1:.*]]
; CHECK-NEXT: i8 0, label %[[MEMCHR_CASE2:.*]]
; CHECK-NEXT: i8 50, label %[[MEMCHR_CASE3:.*]]
; CHECK-NEXT: ], !prof [[PROF1:![0-9]+]]
; CHECK: [[MEMCHR_CASE]]:
; CHECK-NEXT: br label %[[MEMCHR_SUCCESS:.*]]
; CHECK: [[MEMCHR_CASE1]]:
; CHECK-NEXT: br label %[[MEMCHR_SUCCESS]]
; CHECK: [[MEMCHR_CASE2]]:
; CHECK-NEXT: br label %[[MEMCHR_SUCCESS]]
; CHECK: [[MEMCHR_CASE3]]:
; CHECK-NEXT: br label %[[MEMCHR_SUCCESS]]
; CHECK: [[MEMCHR_SUCCESS]]:
; CHECK-NEXT: [[MEMCHR_IDX:%.*]] = phi i64 [ 0, %[[MEMCHR_CASE]] ], [ 1, %[[MEMCHR_CASE1]] ], [ 2, %[[MEMCHR_CASE2]] ], [ 3, %[[MEMCHR_CASE3]] ]
; CHECK-NEXT: [[TMP1:%.*]] = getelementptr inbounds i8, ptr @str, i64 [[MEMCHR_IDX]]
; CHECK-NEXT: br label %[[ENTRY_SPLIT]]
; CHECK: [[ENTRY_SPLIT]]:
; CHECK-NEXT: [[MEMCHR4:%.*]] = phi ptr [ null, %[[ENTRY]] ], [ [[TMP1]], %[[MEMCHR_SUCCESS]] ]
; CHECK-NEXT: [[ISNULL:%.*]] = icmp eq ptr [[MEMCHR4]], null
; CHECK-NEXT: ret i1 [[ISNULL]]
;
entry:
%memchr = call ptr @memchr(ptr @str, i32 %x, i64 5)
%isnull = icmp eq ptr %memchr, null
ret i1 %isnull
}
define ptr @test_memchr(i32 %x) !prof !0 {
; CHECK-LABEL: define ptr @test_memchr(
; CHECK-SAME: i32 [[X:%.*]]) !prof [[PROF0]] {
; CHECK-NEXT: [[ENTRY:.*]]:
; CHECK-NEXT: [[TMP0:%.*]] = trunc i32 [[X]] to i8
; CHECK-NEXT: switch i8 [[TMP0]], label %[[ENTRY_SPLIT:.*]] [
; CHECK-NEXT: i8 48, label %[[MEMCHR_CASE:.*]]
; CHECK-NEXT: i8 49, label %[[MEMCHR_CASE1:.*]]
; CHECK-NEXT: i8 0, label %[[MEMCHR_CASE2:.*]]
; CHECK-NEXT: i8 50, label %[[MEMCHR_CASE3:.*]]
; CHECK-NEXT: ], !prof [[PROF1]]
; CHECK: [[MEMCHR_CASE]]:
; CHECK-NEXT: br label %[[MEMCHR_SUCCESS:.*]]
; CHECK: [[MEMCHR_CASE1]]:
; CHECK-NEXT: br label %[[MEMCHR_SUCCESS]]
; CHECK: [[MEMCHR_CASE2]]:
; CHECK-NEXT: br label %[[MEMCHR_SUCCESS]]
; CHECK: [[MEMCHR_CASE3]]:
; CHECK-NEXT: br label %[[MEMCHR_SUCCESS]]
; CHECK: [[MEMCHR_SUCCESS]]:
; CHECK-NEXT: [[MEMCHR_IDX:%.*]] = phi i64 [ 0, %[[MEMCHR_CASE]] ], [ 1, %[[MEMCHR_CASE1]] ], [ 2, %[[MEMCHR_CASE2]] ], [ 3, %[[MEMCHR_CASE3]] ]
; CHECK-NEXT: [[TMP1:%.*]] = getelementptr inbounds i8, ptr @str, i64 [[MEMCHR_IDX]]
; CHECK-NEXT: br label %[[ENTRY_SPLIT]]
; CHECK: [[ENTRY_SPLIT]]:
; CHECK-NEXT: [[MEMCHR4:%.*]] = phi ptr [ null, %[[ENTRY]] ], [ [[TMP1]], %[[MEMCHR_SUCCESS]] ]
; CHECK-NEXT: ret ptr [[MEMCHR4]]
;
entry:
%memchr = call ptr @memchr(ptr @str, i32 %x, i64 5)
ret ptr %memchr
}
define ptr @test_memchr_smaller_n(i32 %x) !prof !0 {
; CHECK-LABEL: define ptr @test_memchr_smaller_n(
; CHECK-SAME: i32 [[X:%.*]]) !prof [[PROF0]] {
; CHECK-NEXT: [[ENTRY:.*]]:
; CHECK-NEXT: [[TMP0:%.*]] = trunc i32 [[X]] to i8
; CHECK-NEXT: switch i8 [[TMP0]], label %[[ENTRY_SPLIT:.*]] [
; CHECK-NEXT: i8 48, label %[[MEMCHR_CASE:.*]]
; CHECK-NEXT: i8 49, label %[[MEMCHR_CASE1:.*]]
; CHECK-NEXT: i8 0, label %[[MEMCHR_CASE2:.*]]
; CHECK-NEXT: ], !prof [[PROF1]]
; CHECK: [[MEMCHR_CASE]]:
; CHECK-NEXT: br label %[[MEMCHR_SUCCESS:.*]]
; CHECK: [[MEMCHR_CASE1]]:
; CHECK-NEXT: br label %[[MEMCHR_SUCCESS]]
; CHECK: [[MEMCHR_CASE2]]:
; CHECK-NEXT: br label %[[MEMCHR_SUCCESS]]
; CHECK: [[MEMCHR_SUCCESS]]:
; CHECK-NEXT: [[MEMCHR_IDX:%.*]] = phi i64 [ 0, %[[MEMCHR_CASE]] ], [ 1, %[[MEMCHR_CASE1]] ], [ 2, %[[MEMCHR_CASE2]] ]
; CHECK-NEXT: [[TMP1:%.*]] = getelementptr inbounds i8, ptr @str, i64 [[MEMCHR_IDX]]
; CHECK-NEXT: br label %[[ENTRY_SPLIT]]
; CHECK: [[ENTRY_SPLIT]]:
; CHECK-NEXT: [[MEMCHR3:%.*]] = phi ptr [ null, %[[ENTRY]] ], [ [[TMP1]], %[[MEMCHR_SUCCESS]] ]
; CHECK-NEXT: ret ptr [[MEMCHR3]]
;
entry:
%memchr = call ptr @memchr(ptr @str, i32 %x, i64 3)
ret ptr %memchr
}
; negative tests
define ptr @test_memchr_larger_n(i32 %x) !prof !0 {
; CHECK-LABEL: define ptr @test_memchr_larger_n(
; CHECK-SAME: i32 [[X:%.*]]) !prof [[PROF0]] {
; CHECK-NEXT: [[ENTRY:.*:]]
; CHECK-NEXT: [[MEMCHR:%.*]] = call ptr @memchr(ptr @str, i32 [[X]], i64 6)
; CHECK-NEXT: ret ptr [[MEMCHR]]
;
entry:
%memchr = call ptr @memchr(ptr @str, i32 %x, i64 6)
ret ptr %memchr
}
define ptr @test_memchr_non_constant(i32 %x, ptr %str) !prof !0 {
; CHECK-LABEL: define ptr @test_memchr_non_constant(
; CHECK-SAME: i32 [[X:%.*]], ptr [[STR:%.*]]) !prof [[PROF0]] {
; CHECK-NEXT: [[ENTRY:.*:]]
; CHECK-NEXT: [[MEMCHR:%.*]] = call ptr @memchr(ptr [[STR]], i32 [[X]], i64 5)
; CHECK-NEXT: ret ptr [[MEMCHR]]
;
entry:
%memchr = call ptr @memchr(ptr %str, i32 %x, i64 5)
ret ptr %memchr
}
define ptr @test_memchr_constant_ch() !prof !0 {
; CHECK-LABEL: define ptr @test_memchr_constant_ch(
; CHECK-SAME: ) !prof [[PROF0]] {
; CHECK-NEXT: [[ENTRY:.*:]]
; CHECK-NEXT: [[MEMCHR:%.*]] = call ptr @memchr(ptr @str, i32 49, i64 5)
; CHECK-NEXT: ret ptr [[MEMCHR]]
;
entry:
%memchr = call ptr @memchr(ptr @str, i32 49, i64 5)
ret ptr %memchr
}
define ptr @test_memchr_dynamic_n(i32 %x, i32 %y) !prof !0 {
; CHECK-LABEL: define ptr @test_memchr_dynamic_n(
; CHECK-SAME: i32 [[X:%.*]], i32 [[Y:%.*]]) !prof [[PROF0]] {
; CHECK-NEXT: [[ENTRY:.*:]]
; CHECK-NEXT: [[MEMCHR:%.*]] = call ptr @memchr(ptr @str, i32 [[X]], i32 [[Y]])
; CHECK-NEXT: ret ptr [[MEMCHR]]
;
entry:
%memchr = call ptr @memchr(ptr @str, i32 %x, i32 %y)
ret ptr %memchr
}
define ptr @test_memchr_long(i32 %x) !prof !0 {
; CHECK-LABEL: define ptr @test_memchr_long(
; CHECK-SAME: i32 [[X:%.*]]) !prof [[PROF0]] {
; CHECK-NEXT: [[ENTRY:.*:]]
; CHECK-NEXT: [[MEMCHR:%.*]] = call ptr @memchr(ptr @str_long, i32 [[X]], i64 8)
; CHECK-NEXT: ret ptr [[MEMCHR]]
;
entry:
%memchr = call ptr @memchr(ptr @str_long, i32 %x, i64 8)
ret ptr %memchr
}
; We want to check that the compiler still calls memchr if the length is non-constant:
define ptr @test_memchr_non_constant_length2(i32 %x, i64 %len) !prof !0 {
; CHECK-LABEL: define ptr @test_memchr_non_constant_length2(
; CHECK-SAME: i32 [[X:%.*]], i64 [[LEN:%.*]]) !prof [[PROF0]] {
; CHECK-NEXT: [[ENTRY:.*:]]
; CHECK-NEXT: [[MEMCHR:%.*]] = call ptr @memchr(ptr @str, i32 [[X]], i64 [[LEN]])
; CHECK-NEXT: ret ptr [[MEMCHR]]
;
entry:
%memchr = call ptr @memchr(ptr @str, i32 %x, i64 %len)
ret ptr %memchr
}
define ptr @test_memchr_high_bytes(i32 %c) {
; CHECK-LABEL: define ptr @test_memchr_high_bytes(
; CHECK-SAME: i32 [[C:%.*]]) {
; CHECK-NEXT: [[TMP1:%.*]] = trunc i32 [[C]] to i8
; CHECK-NEXT: switch i8 [[TMP1]], [[DOTSPLIT:label %.*]] [
; CHECK-NEXT: i8 -128, label %[[MEMCHR_CASE:.*]]
; CHECK-NEXT: i8 -127, label %[[MEMCHR_CASE1:.*]]
; CHECK-NEXT: i8 0, label %[[MEMCHR_CASE2:.*]]
; CHECK-NEXT: ]
; CHECK: [[MEMCHR_CASE]]:
; CHECK-NEXT: br label %[[MEMCHR_SUCCESS:.*]]
; CHECK: [[MEMCHR_CASE1]]:
; CHECK-NEXT: br label %[[MEMCHR_SUCCESS]]
; CHECK: [[MEMCHR_CASE2]]:
; CHECK-NEXT: br label %[[MEMCHR_SUCCESS]]
; CHECK: [[MEMCHR_SUCCESS]]:
; CHECK-NEXT: [[MEMCHR_IDX:%.*]] = phi i64 [ 0, %[[MEMCHR_CASE]] ], [ 1, %[[MEMCHR_CASE1]] ], [ 2, %[[MEMCHR_CASE2]] ]
; CHECK-NEXT: [[TMP2:%.*]] = getelementptr inbounds i8, ptr @str_hi, i64 [[MEMCHR_IDX]]
; CHECK-NEXT: br [[DOTSPLIT]]
; CHECK: [[_SPLIT:.*:]]
; CHECK-NEXT: [[MEMCHR3:%.*]] = phi ptr [ null, [[TMP0:%.*]] ], [ [[TMP2]], %[[MEMCHR_SUCCESS]] ]
; CHECK-NEXT: ret ptr [[MEMCHR3]]
;
%memchr = call ptr @memchr(ptr @str_hi, i32 %c, i64 3)
ret ptr %memchr
}
!0 = !{!"function_entry_count", i64 1000}
;.
; CHECK: [[PROF0]] = !{!"function_entry_count", i64 1000}
; CHECK: [[PROF1]] = !{!"unknown", !"aggressive-instcombine"}
;.
Reference in New Issue View Git Blame Copy Permalink