[llvm] 04f61fb - [LICM] Add tests for GEP reassociation (NFC)
Nikita Popov via llvm-commits
llvm-commits at lists.llvm.org
Fri Mar 24 07:25:06 PDT 2023
Author: Nikita Popov
Date: 2023-03-24T15:24:58+01:00
New Revision: 04f61fb73dc6a994ab267d431f2fdaedc67430ff
URL: https://github.com/llvm/llvm-project/commit/04f61fb73dc6a994ab267d431f2fdaedc67430ff
DIFF: https://github.com/llvm/llvm-project/commit/04f61fb73dc6a994ab267d431f2fdaedc67430ff.diff
LOG: [LICM] Add tests for GEP reassociation (NFC)
Added:
llvm/test/Transforms/LICM/gep-reassociate.ll
Modified:
Removed:
################################################################################
diff --git a/llvm/test/Transforms/LICM/gep-reassociate.ll b/llvm/test/Transforms/LICM/gep-reassociate.ll
new file mode 100644
index 0000000000000..4283eb11fe9f3
--- /dev/null
+++ b/llvm/test/Transforms/LICM/gep-reassociate.ll
@@ -0,0 +1,333 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2
+; RUN: opt -S -passes=licm < %s | FileCheck %s
+
+declare void @use(ptr)
+declare i32 @get.i32()
+declare i64 @get.i64()
+declare ptr @get.ptr()
+
+define void @only_one_inbounds(ptr %ptr, i1 %c, i32 %arg) {
+; CHECK-LABEL: define void @only_one_inbounds
+; CHECK-SAME: (ptr [[PTR:%.*]], i1 [[C:%.*]], i32 [[ARG:%.*]]) {
+; CHECK-NEXT: entry:
+; CHECK-NEXT: [[ARG_EXT:%.*]] = zext i32 [[ARG]] to i64
+; CHECK-NEXT: br label [[LOOP:%.*]]
+; CHECK: loop:
+; CHECK-NEXT: [[VAL:%.*]] = call i32 @get.i32()
+; CHECK-NEXT: [[VAL_EXT:%.*]] = zext i32 [[VAL]] to i64
+; CHECK-NEXT: [[PTR2:%.*]] = getelementptr inbounds i8, ptr [[PTR]], i64 [[VAL_EXT]]
+; CHECK-NEXT: [[PTR3:%.*]] = getelementptr i8, ptr [[PTR2]], i64 [[ARG_EXT]]
+; CHECK-NEXT: call void @use(ptr [[PTR3]])
+; CHECK-NEXT: br i1 [[C]], label [[LOOP]], label [[EXIT:%.*]]
+; CHECK: exit:
+; CHECK-NEXT: ret void
+;
+entry:
+ %arg.ext = zext i32 %arg to i64
+ br label %loop
+
+loop:
+ %val = call i32 @get.i32()
+ %val.ext = zext i32 %val to i64
+ %ptr2 = getelementptr inbounds i8, ptr %ptr, i64 %val.ext
+ %ptr3 = getelementptr i8, ptr %ptr2, i64 %arg.ext
+ call void @use(ptr %ptr3)
+ br i1 %c, label %loop, label %exit
+
+exit:
+ ret void
+}
+
+define void @both_inbounds_one_neg(ptr %ptr, i1 %c) {
+; CHECK-LABEL: define void @both_inbounds_one_neg
+; CHECK-SAME: (ptr [[PTR:%.*]], i1 [[C:%.*]]) {
+; CHECK-NEXT: entry:
+; CHECK-NEXT: br label [[LOOP:%.*]]
+; CHECK: loop:
+; CHECK-NEXT: [[VAL:%.*]] = call i32 @get.i32()
+; CHECK-NEXT: [[VAL_EXT:%.*]] = zext i32 [[VAL]] to i64
+; CHECK-NEXT: [[PTR2:%.*]] = getelementptr inbounds i8, ptr [[PTR]], i64 [[VAL_EXT]]
+; CHECK-NEXT: [[PTR3:%.*]] = getelementptr i8, ptr [[PTR2]], i64 -1
+; CHECK-NEXT: call void @use(ptr [[PTR3]])
+; CHECK-NEXT: br i1 [[C]], label [[LOOP]], label [[EXIT:%.*]]
+; CHECK: exit:
+; CHECK-NEXT: ret void
+;
+entry:
+ br label %loop
+
+loop:
+ %val = call i32 @get.i32()
+ %val.ext = zext i32 %val to i64
+ %ptr2 = getelementptr inbounds i8, ptr %ptr, i64 %val.ext
+ %ptr3 = getelementptr i8, ptr %ptr2, i64 -1
+ call void @use(ptr %ptr3)
+ br i1 %c, label %loop, label %exit
+
+exit:
+ ret void
+}
+
+define void @both_inbounds_pos(ptr %ptr, i1 %c) {
+; CHECK-LABEL: define void @both_inbounds_pos
+; CHECK-SAME: (ptr [[PTR:%.*]], i1 [[C:%.*]]) {
+; CHECK-NEXT: entry:
+; CHECK-NEXT: br label [[LOOP:%.*]]
+; CHECK: loop:
+; CHECK-NEXT: [[VAL:%.*]] = call i32 @get.i32()
+; CHECK-NEXT: [[VAL_EXT:%.*]] = zext i32 [[VAL]] to i64
+; CHECK-NEXT: [[PTR2:%.*]] = getelementptr inbounds i8, ptr [[PTR]], i64 [[VAL_EXT]]
+; CHECK-NEXT: [[PTR3:%.*]] = getelementptr inbounds i8, ptr [[PTR2]], i64 1
+; CHECK-NEXT: call void @use(ptr [[PTR3]])
+; CHECK-NEXT: br i1 [[C]], label [[LOOP]], label [[EXIT:%.*]]
+; CHECK: exit:
+; CHECK-NEXT: ret void
+;
+entry:
+ br label %loop
+
+loop:
+ %val = call i32 @get.i32()
+ %val.ext = zext i32 %val to i64
+ %ptr2 = getelementptr inbounds i8, ptr %ptr, i64 %val.ext
+ %ptr3 = getelementptr inbounds i8, ptr %ptr2, i64 1
+ call void @use(ptr %ptr3)
+ br i1 %c, label %loop, label %exit
+
+exit:
+ ret void
+}
+
+define void @
diff erent_elem_types(ptr %ptr, i1 %c, i64 %arg) {
+; CHECK-LABEL: define void @
diff erent_elem_types
+; CHECK-SAME: (ptr [[PTR:%.*]], i1 [[C:%.*]], i64 [[ARG:%.*]]) {
+; CHECK-NEXT: entry:
+; CHECK-NEXT: br label [[LOOP:%.*]]
+; CHECK: loop:
+; CHECK-NEXT: [[VAL:%.*]] = call i64 @get.i64()
+; CHECK-NEXT: [[PTR2:%.*]] = getelementptr i32, ptr [[PTR]], i64 [[VAL]]
+; CHECK-NEXT: [[PTR3:%.*]] = getelementptr i64, ptr [[PTR2]], i64 [[ARG]]
+; CHECK-NEXT: call void @use(ptr [[PTR3]])
+; CHECK-NEXT: br i1 [[C]], label [[LOOP]], label [[EXIT:%.*]]
+; CHECK: exit:
+; CHECK-NEXT: ret void
+;
+entry:
+ br label %loop
+
+loop:
+ %val = call i64 @get.i64()
+ %ptr2 = getelementptr i32, ptr %ptr, i64 %val
+ %ptr3 = getelementptr i64, ptr %ptr2, i64 %arg
+ call void @use(ptr %ptr3)
+ br i1 %c, label %loop, label %exit
+
+exit:
+ ret void
+}
+
+define void @src_has_extra_use(ptr %ptr, i1 %c, i64 %arg) {
+; CHECK-LABEL: define void @src_has_extra_use
+; CHECK-SAME: (ptr [[PTR:%.*]], i1 [[C:%.*]], i64 [[ARG:%.*]]) {
+; CHECK-NEXT: entry:
+; CHECK-NEXT: br label [[LOOP:%.*]]
+; CHECK: loop:
+; CHECK-NEXT: [[VAL:%.*]] = call i64 @get.i64()
+; CHECK-NEXT: [[PTR2:%.*]] = getelementptr i8, ptr [[PTR]], i64 [[VAL]]
+; CHECK-NEXT: call void @use(ptr [[PTR2]])
+; CHECK-NEXT: [[PTR3:%.*]] = getelementptr i8, ptr [[PTR2]], i64 [[ARG]]
+; CHECK-NEXT: call void @use(ptr [[PTR3]])
+; CHECK-NEXT: br i1 [[C]], label [[LOOP]], label [[EXIT:%.*]]
+; CHECK: exit:
+; CHECK-NEXT: ret void
+;
+entry:
+ br label %loop
+
+loop:
+ %val = call i64 @get.i64()
+ %ptr2 = getelementptr i8, ptr %ptr, i64 %val
+ call void @use(ptr %ptr2)
+ %ptr3 = getelementptr i8, ptr %ptr2, i64 %arg
+ call void @use(ptr %ptr3)
+ br i1 %c, label %loop, label %exit
+
+exit:
+ ret void
+}
+
+define void @src_already_invariant(ptr %ptr, i1 %c, i64 %arg1, i64 %arg2) {
+; CHECK-LABEL: define void @src_already_invariant
+; CHECK-SAME: (ptr [[PTR:%.*]], i1 [[C:%.*]], i64 [[ARG1:%.*]], i64 [[ARG2:%.*]]) {
+; CHECK-NEXT: entry:
+; CHECK-NEXT: [[PTR2:%.*]] = getelementptr i8, ptr [[PTR]], i64 [[ARG1]]
+; CHECK-NEXT: [[PTR3:%.*]] = getelementptr i8, ptr [[PTR2]], i64 [[ARG2]]
+; CHECK-NEXT: br label [[LOOP:%.*]]
+; CHECK: loop:
+; CHECK-NEXT: call void @use(ptr [[PTR3]])
+; CHECK-NEXT: br i1 [[C]], label [[LOOP]], label [[EXIT:%.*]]
+; CHECK: exit:
+; CHECK-NEXT: ret void
+;
+entry:
+ br label %loop
+
+loop:
+ %ptr2 = getelementptr i8, ptr %ptr, i64 %arg1
+ %ptr3 = getelementptr i8, ptr %ptr2, i64 %arg2
+ call void @use(ptr %ptr3)
+ br i1 %c, label %loop, label %exit
+
+exit:
+ ret void
+}
+
+define void @gep_idx_not_invariant(ptr %ptr, i1 %c) {
+; CHECK-LABEL: define void @gep_idx_not_invariant
+; CHECK-SAME: (ptr [[PTR:%.*]], i1 [[C:%.*]]) {
+; CHECK-NEXT: entry:
+; CHECK-NEXT: br label [[LOOP:%.*]]
+; CHECK: loop:
+; CHECK-NEXT: [[VAL1:%.*]] = call i64 @get.i64()
+; CHECK-NEXT: [[VAL2:%.*]] = call i64 @get.i64()
+; CHECK-NEXT: [[PTR2:%.*]] = getelementptr i8, ptr [[PTR]], i64 [[VAL1]]
+; CHECK-NEXT: [[PTR3:%.*]] = getelementptr i8, ptr [[PTR2]], i64 [[VAL2]]
+; CHECK-NEXT: call void @use(ptr [[PTR3]])
+; CHECK-NEXT: br i1 [[C]], label [[LOOP]], label [[EXIT:%.*]]
+; CHECK: exit:
+; CHECK-NEXT: ret void
+;
+entry:
+ br label %loop
+
+loop:
+ %val1 = call i64 @get.i64()
+ %val2 = call i64 @get.i64()
+ %ptr2 = getelementptr i8, ptr %ptr, i64 %val1
+ %ptr3 = getelementptr i8, ptr %ptr2, i64 %val2
+ call void @use(ptr %ptr3)
+ br i1 %c, label %loop, label %exit
+
+exit:
+ ret void
+}
+
+define void @src_ptr_not_invariant(i1 %c, i64 %arg) {
+; CHECK-LABEL: define void @src_ptr_not_invariant
+; CHECK-SAME: (i1 [[C:%.*]], i64 [[ARG:%.*]]) {
+; CHECK-NEXT: entry:
+; CHECK-NEXT: br label [[LOOP:%.*]]
+; CHECK: loop:
+; CHECK-NEXT: [[VAL:%.*]] = call i64 @get.i64()
+; CHECK-NEXT: [[PTR:%.*]] = call ptr @get.ptr()
+; CHECK-NEXT: [[PTR2:%.*]] = getelementptr i8, ptr [[PTR]], i64 [[ARG]]
+; CHECK-NEXT: [[PTR3:%.*]] = getelementptr i8, ptr [[PTR2]], i64 [[VAL]]
+; CHECK-NEXT: call void @use(ptr [[PTR3]])
+; CHECK-NEXT: br i1 [[C]], label [[LOOP]], label [[EXIT:%.*]]
+; CHECK: exit:
+; CHECK-NEXT: ret void
+;
+entry:
+ br label %loop
+
+loop:
+ %val = call i64 @get.i64()
+ %ptr = call ptr @get.ptr()
+ %ptr2 = getelementptr i8, ptr %ptr, i64 %arg
+ %ptr3 = getelementptr i8, ptr %ptr2, i64 %val
+ call void @use(ptr %ptr3)
+ br i1 %c, label %loop, label %exit
+
+exit:
+ ret void
+}
+
+define void @multiple_indices(ptr %ptr, i1 %c, i64 %arg1, i64 %arg2) {
+; CHECK-LABEL: define void @multiple_indices
+; CHECK-SAME: (ptr [[PTR:%.*]], i1 [[C:%.*]], i64 [[ARG1:%.*]], i64 [[ARG2:%.*]]) {
+; CHECK-NEXT: entry:
+; CHECK-NEXT: br label [[LOOP:%.*]]
+; CHECK: loop:
+; CHECK-NEXT: [[VAL1:%.*]] = call i64 @get.i64()
+; CHECK-NEXT: [[VAL2:%.*]] = call i64 @get.i64()
+; CHECK-NEXT: [[PTR2:%.*]] = getelementptr [0 x i8], ptr [[PTR]], i64 [[VAL1]], i64 [[VAL2]]
+; CHECK-NEXT: [[PTR3:%.*]] = getelementptr [0 x i8], ptr [[PTR2]], i64 [[ARG1]], i64 [[ARG2]]
+; CHECK-NEXT: call void @use(ptr [[PTR3]])
+; CHECK-NEXT: br i1 [[C]], label [[LOOP]], label [[EXIT:%.*]]
+; CHECK: exit:
+; CHECK-NEXT: ret void
+;
+entry:
+ br label %loop
+
+loop:
+ %val1 = call i64 @get.i64()
+ %val2 = call i64 @get.i64()
+ %ptr2 = getelementptr [0 x i8], ptr %ptr, i64 %val1, i64 %val2
+ %ptr3 = getelementptr [0 x i8], ptr %ptr2, i64 %arg1, i64 %arg2
+ call void @use(ptr %ptr3)
+ br i1 %c, label %loop, label %exit
+
+exit:
+ ret void
+}
+
+define void @multiple_indices_not_invariant(ptr %ptr, i1 %c, i64 %arg1) {
+; CHECK-LABEL: define void @multiple_indices_not_invariant
+; CHECK-SAME: (ptr [[PTR:%.*]], i1 [[C:%.*]], i64 [[ARG1:%.*]]) {
+; CHECK-NEXT: entry:
+; CHECK-NEXT: br label [[LOOP:%.*]]
+; CHECK: loop:
+; CHECK-NEXT: [[VAL1:%.*]] = call i64 @get.i64()
+; CHECK-NEXT: [[VAL2:%.*]] = call i64 @get.i64()
+; CHECK-NEXT: [[VAL3:%.*]] = call i64 @get.i64()
+; CHECK-NEXT: [[PTR2:%.*]] = getelementptr [0 x i8], ptr [[PTR]], i64 [[VAL1]], i64 [[VAL2]]
+; CHECK-NEXT: [[PTR3:%.*]] = getelementptr [0 x i8], ptr [[PTR2]], i64 [[ARG1]], i64 [[VAL3]]
+; CHECK-NEXT: call void @use(ptr [[PTR3]])
+; CHECK-NEXT: br i1 [[C]], label [[LOOP]], label [[EXIT:%.*]]
+; CHECK: exit:
+; CHECK-NEXT: ret void
+;
+entry:
+ br label %loop
+
+loop:
+ %val1 = call i64 @get.i64()
+ %val2 = call i64 @get.i64()
+ %val3 = call i64 @get.i64()
+ %ptr2 = getelementptr [0 x i8], ptr %ptr, i64 %val1, i64 %val2
+ %ptr3 = getelementptr [0 x i8], ptr %ptr2, i64 %arg1, i64 %val3
+ call void @use(ptr %ptr3)
+ br i1 %c, label %loop, label %exit
+
+exit:
+ ret void
+}
+
+define void @multiple_indices_very_invariant(ptr %ptr, i1 %c, i64 %arg1, i64 %arg2, i64 %arg3) {
+; CHECK-LABEL: define void @multiple_indices_very_invariant
+; CHECK-SAME: (ptr [[PTR:%.*]], i1 [[C:%.*]], i64 [[ARG1:%.*]], i64 [[ARG2:%.*]], i64 [[ARG3:%.*]]) {
+; CHECK-NEXT: entry:
+; CHECK-NEXT: br label [[LOOP:%.*]]
+; CHECK: loop:
+; CHECK-NEXT: [[VAL1:%.*]] = call i64 @get.i64()
+; CHECK-NEXT: [[PTR2:%.*]] = getelementptr [0 x i8], ptr [[PTR]], i64 [[ARG3]], i64 [[VAL1]]
+; CHECK-NEXT: [[PTR3:%.*]] = getelementptr [0 x i8], ptr [[PTR2]], i64 [[ARG1]], i64 [[ARG2]]
+; CHECK-NEXT: call void @use(ptr [[PTR3]])
+; CHECK-NEXT: br i1 [[C]], label [[LOOP]], label [[EXIT:%.*]]
+; CHECK: exit:
+; CHECK-NEXT: ret void
+;
+entry:
+ br label %loop
+
+loop:
+ %val1 = call i64 @get.i64()
+ %ptr2 = getelementptr [0 x i8], ptr %ptr, i64 %arg3, i64 %val1
+ %ptr3 = getelementptr [0 x i8], ptr %ptr2, i64 %arg1, i64 %arg2
+ call void @use(ptr %ptr3)
+ br i1 %c, label %loop, label %exit
+
+exit:
+ ret void
+}
More information about the llvm-commits
mailing list