xref: /llvm-project/llvm/test/Transforms/LoopPredication/unswitch-exit-loop.ll (revision ba89c66771a8512ddbee7391e8ea11f7149f7ae2)
1e23e1594SRoman Lebedev; RUN: opt < %s -passes=loop-predication -S | FileCheck %s
255bdb140SAnna Thomas; RUN: opt -S -passes='require<scalar-evolution>,loop-mssa(loop-predication)' -verify-memoryssa < %s 2>&1 | FileCheck %s
3cc7cb05eSFedor Sergeev
4cc7cb05eSFedor Sergeev;; This is a simplified copy of @unswitch_exit_form test that should trigger loop-predication
5cc7cb05eSFedor Sergeev;; activity and properly bail out when discovering that widenable check does not lead to deopt.
6cc7cb05eSFedor Sergeev;;
7cc7cb05eSFedor Sergeev;; Error checking is rather silly here - it should pass compilation successfully,
8cc7cb05eSFedor Sergeev;; in bad case it will just timeout.
9cc7cb05eSFedor Sergeev;;
10*ba89c667SNikita Popovdefine i32 @unswitch_exit_form_with_endless_loop(ptr %array, i32 %length, i32 %n, i1 %cond_0) {
11cc7cb05eSFedor Sergeev; CHECK-LABEL: @unswitch_exit_form_with_endless_loop
12cc7cb05eSFedor Sergeeventry:
13cc7cb05eSFedor Sergeev  %widenable_cond = call i1 @llvm.experimental.widenable.condition()
14cc7cb05eSFedor Sergeev  %exiplicit_guard_cond = and i1 %cond_0, %widenable_cond
15cc7cb05eSFedor Sergeev  br i1 %exiplicit_guard_cond, label %loop.preheader, label %not_really_a_deopt, !prof !0
16cc7cb05eSFedor Sergeev
17cc7cb05eSFedor Sergeevnot_really_a_deopt:
18cc7cb05eSFedor Sergeev  br label %looping
19cc7cb05eSFedor Sergeev
20cc7cb05eSFedor Sergeevlooping:
21cc7cb05eSFedor Sergeev  ;; synthetic corner case that demonstrates the need for more careful traversal
22cc7cb05eSFedor Sergeev  ;; of unique successors when walking through the exit for profitability checks.
23cc7cb05eSFedor Sergeev  br label %looping
24cc7cb05eSFedor Sergeev
25cc7cb05eSFedor Sergeevloop.preheader:
26cc7cb05eSFedor Sergeev  br label %loop
27cc7cb05eSFedor Sergeev
28cc7cb05eSFedor Sergeevloop:
29cc7cb05eSFedor Sergeev  %loop.acc = phi i32 [ %loop.acc.next, %guarded ], [ 0, %loop.preheader ]
30cc7cb05eSFedor Sergeev  %i = phi i32 [ %i.next, %guarded ], [ 0, %loop.preheader ]
31cc7cb05eSFedor Sergeev  %within.bounds = icmp ult i32 %i, %length
32cc7cb05eSFedor Sergeev  br i1 %within.bounds, label %guarded, label %not_really_a_deopt, !prof !0
33cc7cb05eSFedor Sergeev
34cc7cb05eSFedor Sergeevguarded:
35cc7cb05eSFedor Sergeev  %i.i64 = zext i32 %i to i64
36*ba89c667SNikita Popov  %array.i.ptr = getelementptr inbounds i32, ptr %array, i64 %i.i64
37*ba89c667SNikita Popov  %array.i = load i32, ptr %array.i.ptr, align 4
38*ba89c667SNikita Popov  store i32 0, ptr %array.i.ptr
39cc7cb05eSFedor Sergeev  %loop.acc.next = add i32 %loop.acc, %array.i
40cc7cb05eSFedor Sergeev  %i.next = add nuw i32 %i, 1
41cc7cb05eSFedor Sergeev  %continue = icmp ult i32 %i.next, %n
42cc7cb05eSFedor Sergeev  br i1 %continue, label %loop, label %exit
43cc7cb05eSFedor Sergeev
44cc7cb05eSFedor Sergeevexit:
45cc7cb05eSFedor Sergeev  %result = phi i32 [ %loop.acc.next, %guarded ]
46cc7cb05eSFedor Sergeev  ret i32 %result
47cc7cb05eSFedor Sergeev}
48cc7cb05eSFedor Sergeev
49cc7cb05eSFedor Sergeevdeclare void @unknown()
50cc7cb05eSFedor Sergeev
51cc7cb05eSFedor Sergeevdeclare i1 @llvm.experimental.widenable.condition()
52cc7cb05eSFedor Sergeevdeclare i32 @llvm.experimental.deoptimize.i32(...)
53cc7cb05eSFedor Sergeev
54cc7cb05eSFedor Sergeev!0 = !{!"branch_weights", i32 1048576, i32 1}
55cc7cb05eSFedor Sergeev!1 = !{i32 1, i32 -2147483648}
56cc7cb05eSFedor Sergeev!2 = !{i32 0, i32 50}
57