[llvm] 6ebe394 - [IndVars] Add test for simplifying a induction cmp against global.

[Florian Hahn via llvm-commits]

Author: Florian Hahn
Date: 2023-04-28T20:41:09+01:00
New Revision: 6ebe3949159e18db1022ff2006a9877e3ba92bbf

URL: https://github.com/llvm/llvm-project/commit/6ebe3949159e18db1022ff2006a9877e3ba92bbf
DIFF: https://github.com/llvm/llvm-project/commit/6ebe3949159e18db1022ff2006a9877e3ba92bbf.diff

LOG: [IndVars] Add test for simplifying a induction cmp against global.

Added: 
    llvm/test/Transforms/IndVarSimplify/ptrtoint-global-cmp.ll

Modified: 
    

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/IndVarSimplify/ptrtoint-global-cmp.ll b/llvm/test/Transforms/IndVarSimplify/ptrtoint-global-cmp.ll
new file mode 100644
index 0000000000000..4d43f2a7299e8
--- /dev/null
+++ b/llvm/test/Transforms/IndVarSimplify/ptrtoint-global-cmp.ll
@@ -0,0 +1,58 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2
+; RUN: opt -passes=indvars -S %s | FileCheck %s
+
+ at a = global [4 x i32] zeroinitializer, align 16
+
+define i32 @test() {
+; CHECK-LABEL: define i32 @test() {
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[UMAX:%.*]] = call i64 @llvm.umax.i64(i64 ptrtoint (ptr @a to i64), i64 add (i64 ptrtoint (ptr @a to i64), i64 16))
+; CHECK-NEXT:    [[TMP0:%.*]] = add i64 [[UMAX]], 3
+; CHECK-NEXT:    [[TMP1:%.*]] = sub i64 [[TMP0]], ptrtoint (ptr @a to i64)
+; CHECK-NEXT:    [[TMP2:%.*]] = lshr i64 [[TMP1]], 2
+; CHECK-NEXT:    [[UMIN:%.*]] = call i64 @llvm.umin.i64(i64 [[TMP2]], i64 4)
+; CHECK-NEXT:    [[TMP3:%.*]] = icmp ne i64 [[TMP2]], [[UMIN]]
+; CHECK-NEXT:    br label [[LOOP_HEADER:%.*]]
+; CHECK:       loop.header:
+; CHECK-NEXT:    [[IV:%.*]] = phi i8 [ 0, [[ENTRY:%.*]] ], [ [[IV_NEXT:%.*]], [[LOOP_LATCH:%.*]] ]
+; CHECK-NEXT:    [[RED:%.*]] = phi i32 [ 0, [[ENTRY]] ], [ [[RED_NEXT:%.*]], [[LOOP_LATCH]] ]
+; CHECK-NEXT:    [[IDXPROM:%.*]] = zext i8 [[IV]] to i64
+; CHECK-NEXT:    [[GEP:%.*]] = getelementptr i32, ptr @a, i64 [[IDXPROM]]
+; CHECK-NEXT:    br i1 [[TMP3]], label [[LOOP_LATCH]], label [[EXIT_1:%.*]]
+; CHECK:       loop.latch:
+; CHECK-NEXT:    [[L:%.*]] = load i32, ptr [[GEP]], align 4
+; CHECK-NEXT:    [[RED_NEXT]] = add nsw i32 [[L]], [[RED]]
+; CHECK-NEXT:    [[IV_NEXT]] = add nuw nsw i8 [[IV]], 1
+; CHECK-NEXT:    [[CMP:%.*]] = icmp ult i8 [[IV]], 4
+; CHECK-NEXT:    br i1 [[CMP]], label [[LOOP_HEADER]], label [[EXIT_2:%.*]]
+; CHECK:       exit.1:
+; CHECK-NEXT:    ret i32 0
+; CHECK:       exit.2:
+; CHECK-NEXT:    [[ADD_LCSSA:%.*]] = phi i32 [ [[RED_NEXT]], [[LOOP_LATCH]] ]
+; CHECK-NEXT:    ret i32 [[ADD_LCSSA]]
+;
+entry:
+  br label %loop.header
+
+loop.header:
+  %iv = phi i8 [ 0, %entry ], [ %iv.next, %loop.latch ]
+  %red = phi i32 [ 0, %entry ], [ %red.next, %loop.latch ]
+  %idxprom = zext i8 %iv to i64
+  %gep = getelementptr i32, ptr @a, i64 %idxprom
+  %f = icmp ult ptr %gep, getelementptr inbounds ([4 x i32], ptr @a, i64 1, i64 0)
+  br i1 %f, label %loop.latch, label %exit.1
+
+loop.latch:
+  %l = load i32, ptr %gep, align 4
+  %red.next = add nsw i32 %l, %red
+  %iv.next = add nuw nsw i8 %iv, 1
+  %cmp = icmp ult i8 %iv, 4
+  br i1 %cmp, label %loop.header, label %exit.2
+
+exit.1:                                             ; preds = %for.body
+  ret i32 0
+
+exit.2:
+  %add.lcssa = phi i32 [ %red.next, %loop.latch ]
+  ret i32 %add.lcssa
+}