[llvm] 371d2ed - [InstCombine] Add additional memchr tests (NFC)
Nikita Popov via llvm-commits
llvm-commits at lists.llvm.org
Fri Apr 1 03:19:15 PDT 2022
Author: Martin Sebor
Date: 2022-04-01T12:16:03+02:00
New Revision: 371d2ed3f3d7f67602ff9956510f3e2d9df3d5b0
URL: https://github.com/llvm/llvm-project/commit/371d2ed3f3d7f67602ff9956510f3e2d9df3d5b0
DIFF: https://github.com/llvm/llvm-project/commit/371d2ed3f3d7f67602ff9956510f3e2d9df3d5b0.diff
LOG: [InstCombine] Add additional memchr tests (NFC)
Added:
llvm/test/Transforms/InstCombine/memchr-2.ll
llvm/test/Transforms/InstCombine/memchr-3.ll
Modified:
Removed:
################################################################################
diff --git a/llvm/test/Transforms/InstCombine/memchr-2.ll b/llvm/test/Transforms/InstCombine/memchr-2.ll
new file mode 100644
index 0000000000000..54061fe219d29
--- /dev/null
+++ b/llvm/test/Transforms/InstCombine/memchr-2.ll
@@ -0,0 +1,119 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt < %s -passes=instcombine -S | FileCheck %s
+
+; Verify that memchr calls with constant arrays, or constant characters,
+; or constant bounds are folded (or not) as expected.
+
+declare i8* @memchr(i8*, i32, i64)
+
+ at ax = external global [0 x i8]
+ at a12345 = constant [5 x i8] c"\01\02\03\04\05"
+
+
+; Fold memchr(a12345, '\06', n) to null.
+
+define i8* @fold_memchr_a12345_6_n(i64 %n) {
+; CHECK-LABEL: @fold_memchr_a12345_6_n(
+; CHECK-NEXT: [[RES:%.*]] = call i8* @memchr(i8* getelementptr inbounds ([5 x i8], [5 x i8]* @a12345, i64 0, i64 0), i32 6, i64 [[N:%.*]])
+; CHECK-NEXT: ret i8* [[RES]]
+;
+
+ %ptr = getelementptr [5 x i8], [5 x i8]* @a12345, i32 0, i32 0
+ %res = call i8* @memchr(i8* %ptr, i32 6, i64 %n)
+ ret i8* %res
+}
+
+
+; Fold memchr(a12345, '\04', 2) to null.
+
+define i8* @fold_memchr_a12345_4_2() {
+; CHECK-LABEL: @fold_memchr_a12345_4_2(
+; CHECK-NEXT: ret i8* null
+;
+
+ %ptr = getelementptr [5 x i8], [5 x i8]* @a12345, i32 0, i32 0
+ %res = call i8* @memchr(i8* %ptr, i32 4, i64 2)
+ ret i8* %res
+}
+
+
+; Fold memchr(a12345, '\04', 3) to null.
+
+define i8* @fold_memchr_a12345_4_3() {
+; CHECK-LABEL: @fold_memchr_a12345_4_3(
+; CHECK-NEXT: ret i8* null
+;
+
+ %ptr = getelementptr [5 x i8], [5 x i8]* @a12345, i32 0, i32 0
+ %res = call i8* @memchr(i8* %ptr, i32 4, i64 3)
+ ret i8* %res
+}
+
+
+; Fold memchr(a12345, '\03', 3) to a12345 + 2.
+
+define i8* @fold_memchr_a12345_3_3() {
+; CHECK-LABEL: @fold_memchr_a12345_3_3(
+; CHECK-NEXT: ret i8* getelementptr inbounds ([5 x i8], [5 x i8]* @a12345, i64 0, i64 2)
+;
+
+ %ptr = getelementptr [5 x i8], [5 x i8]* @a12345, i32 0, i32 0
+ %res = call i8* @memchr(i8* %ptr, i32 3, i64 3)
+ ret i8* %res
+}
+
+
+; Fold memchr(a12345, '\03', 9) to a12345 + 2.
+
+define i8* @fold_memchr_a12345_3_9() {
+; CHECK-LABEL: @fold_memchr_a12345_3_9(
+; CHECK-NEXT: ret i8* getelementptr inbounds ([5 x i8], [5 x i8]* @a12345, i64 0, i64 2)
+;
+
+ %ptr = getelementptr [5 x i8], [5 x i8]* @a12345, i32 0, i32 0
+ %res = call i8* @memchr(i8* %ptr, i32 3, i64 9)
+ ret i8* %res
+}
+
+
+; Fold memchr(a12345, '\03', n) to n < 3 ? null : a12345 + 3.
+
+define i8* @call_a12345_3_n(i64 %n) {
+; CHECK-LABEL: @call_a12345_3_n(
+; CHECK-NEXT: [[RES:%.*]] = call i8* @memchr(i8* getelementptr inbounds ([5 x i8], [5 x i8]* @a12345, i64 0, i64 0), i32 3, i64 [[N:%.*]])
+; CHECK-NEXT: ret i8* [[RES]]
+;
+
+ %ptr = getelementptr [5 x i8], [5 x i8]* @a12345, i32 0, i32 0
+ %res = call i8* @memchr(i8* %ptr, i32 3, i64 %n)
+ ret i8* %res
+}
+
+
+; Fold memchr(a12345, 259, n) to n < 4 ? null : a12345 + 3
+; to verify the constant 259 is converted to unsigned char (yielding 3).
+
+define i8* @call_a12345_259_n(i64 %n) {
+; CHECK-LABEL: @call_a12345_259_n(
+; CHECK-NEXT: [[RES:%.*]] = call i8* @memchr(i8* getelementptr inbounds ([5 x i8], [5 x i8]* @a12345, i64 0, i64 0), i32 259, i64 [[N:%.*]])
+; CHECK-NEXT: ret i8* [[RES]]
+;
+
+ %ptr = getelementptr [5 x i8], [5 x i8]* @a12345, i32 0, i32 0
+ %res = call i8* @memchr(i8* %ptr, i32 259, i64 %n)
+ ret i8* %res
+}
+
+
+; Do no fold memchr(ax, 1, n).
+
+define i8* @call_ax_1_n(i64 %n) {
+; CHECK-LABEL: @call_ax_1_n(
+; CHECK-NEXT: [[RES:%.*]] = call i8* @memchr(i8* getelementptr inbounds ([0 x i8], [0 x i8]* @ax, i64 0, i64 0), i32 1, i64 [[N:%.*]])
+; CHECK-NEXT: ret i8* [[RES]]
+;
+
+ %ptr = getelementptr [0 x i8], [0 x i8]* @ax, i32 0, i32 0
+ %res = call i8* @memchr(i8* %ptr, i32 1, i64 %n)
+ ret i8* %res
+}
diff --git a/llvm/test/Transforms/InstCombine/memchr-3.ll b/llvm/test/Transforms/InstCombine/memchr-3.ll
new file mode 100644
index 0000000000000..03094b7b9f631
--- /dev/null
+++ b/llvm/test/Transforms/InstCombine/memchr-3.ll
@@ -0,0 +1,65 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt < %s -passes=instcombine -S | FileCheck %s
+
+; Verify that the special case of memchr calls with the size of 1 are
+; folded as expected.
+
+declare i8* @memchr(i8*, i32, i64)
+
+ at ax = external global [0 x i8]
+ at a12345 = constant [5 x i8] c"\01\02\03\04\05"
+
+
+; Fold memchr(a12345, 1, 1) to a12345.
+
+define i8* @fold_memchr_a12345_1_1() {
+; CHECK-LABEL: @fold_memchr_a12345_1_1(
+; CHECK-NEXT: ret i8* getelementptr inbounds ([5 x i8], [5 x i8]* @a12345, i64 0, i64 0)
+;
+
+ %ptr = getelementptr [5 x i8], [5 x i8]* @a12345, i32 0, i32 0
+ %res = call i8* @memchr(i8* %ptr, i32 1, i64 1)
+ ret i8* %res
+}
+
+
+; Fold memchr(a12345, 2, 1) to null.
+
+define i8* @fold_memchr_a12345_2_1() {
+; CHECK-LABEL: @fold_memchr_a12345_2_1(
+; CHECK-NEXT: ret i8* null
+;
+
+ %ptr = getelementptr [5 x i8], [5 x i8]* @a12345, i32 0, i32 0
+ %res = call i8* @memchr(i8* %ptr, i32 2, i64 1)
+ ret i8* %res
+}
+
+
+; Fold memchr(ax, 257, 1) to (unsigned char)*ax == 1 ? ax : null
+; to verify the constant 257 is converted to unsigned char (yielding 1).
+
+define i8* @fold_memchr_ax_257_1(i32 %chr, i64 %n) {
+; CHECK-LABEL: @fold_memchr_ax_257_1(
+; CHECK-NEXT: [[RES:%.*]] = call i8* @memchr(i8* noundef nonnull dereferenceable(1) getelementptr inbounds ([0 x i8], [0 x i8]* @ax, i64 0, i64 0), i32 257, i64 1)
+; CHECK-NEXT: ret i8* [[RES]]
+;
+
+ %ptr = getelementptr [0 x i8], [0 x i8]* @ax, i32 0, i32 0
+ %res = call i8* @memchr(i8* %ptr, i32 257, i64 1)
+ ret i8* %res
+}
+
+
+; Fold memchr(ax, c, 1) to (unsigned char)*ax == (unsigned char)c ? ax : null.
+
+define i8* @fold_memchr_ax_c_1(i32 %chr, i64 %n) {
+; CHECK-LABEL: @fold_memchr_ax_c_1(
+; CHECK-NEXT: [[RES:%.*]] = call i8* @memchr(i8* noundef nonnull dereferenceable(1) getelementptr inbounds ([0 x i8], [0 x i8]* @ax, i64 0, i64 0), i32 [[CHR:%.*]], i64 1)
+; CHECK-NEXT: ret i8* [[RES]]
+;
+
+ %ptr = getelementptr [0 x i8], [0 x i8]* @ax, i32 0, i32 0
+ %res = call i8* @memchr(i8* %ptr, i32 %chr, i64 1)
+ ret i8* %res
+}
More information about the llvm-commits
mailing list