xref: /llvm-project/flang/test/Lower/goto-do-body.f90 (revision a88677edc0792534ba3157bf7d7a1b98e470f2fb)
1! RUN: bbc %s -emit-fir -hlfir=false -o - | FileCheck %s
2
3! Test jumping to the body of a do loop.
4subroutine sub1()
5! CHECK-LABEL:  func @_QPsub1() {
6  implicit none
7  integer :: i
8  external foo
9! CHECK:    %[[TRIP:.*]] = fir.alloca i32
10! CHECK:    %[[I:.*]] = fir.alloca i32 {bindc_name = "i", {{.*}}}
11
12  do i = 1, 3
13    if (i .eq. 2) goto 70
14! CHECK:    %[[TMP1:.*]] = fir.load %[[I]] : !fir.ref<i32>
15! CHECK:    %[[C2_1:.*]] = arith.constant 2 : i32
16! CHECK:    %[[COND:.*]] = arith.cmpi eq, %[[TMP1]], %[[C2_1]] : i32
17! CHECK:    cf.cond_br %[[COND]], ^[[COND_TRUE:.*]], ^{{.*}}
18! CHECK:  ^[[COND_TRUE]]:
19! CHECK:    cf.br ^[[BODY:.*]]
20
21  end do
22
23  call foo
24! CHECK:  fir.call @_QPfoo()
25
26  do i = 1, 2
27! CHECK:    %[[C1_1:.*]] = arith.constant 1 : i32
28! CHECK:    %[[C2_2:.*]] = arith.constant 2 : i32
29! CHECK:    %[[C1_2:.*]] = arith.constant 1 : i32
30! CHECK:    %[[TMP2:.*]] = arith.subi %[[C2_2]], %[[C1_1]] : i32
31! CHECK:    %[[TMP3:.*]] = arith.addi %[[TMP2]], %[[C1_2]] : i32
32! CHECK:    %[[TMP4:.*]] = arith.divsi %[[TMP3]], %[[C1_2]] : i32
33! CHECK:    fir.store %[[TMP4]] to %[[TRIP]] : !fir.ref<i32>
34! CHECK:    fir.store %[[C1_1]] to %[[I]] : !fir.ref<i32>
35! CHECK:    cf.br ^[[HEADER:.*]]
36! CHECK:  ^[[HEADER]]:
37! CHECK:    %[[TMP5:.*]] = fir.load %[[TRIP]] : !fir.ref<i32>
38! CHECK:    %[[C0_1:.*]] = arith.constant 0 : i32
39! CHECK:    %[[TMP6:.*]] = arith.cmpi sgt, %[[TMP5]], %[[C0_1]] : i32
40! CHECK:    cf.cond_br %[[TMP6]], ^[[BODY]], ^[[EXIT:.*]]
41
42    70 call foo
43! CHECK:  ^[[BODY]]:
44! CHECK:    fir.call @_QPfoo()
45! CHECK:    %[[TMP7:.*]] = fir.load %[[TRIP]] : !fir.ref<i32>
46! CHECK:    %[[C1_3:.*]] = arith.constant 1 : i32
47! CHECK:    %[[TMP8:.*]] = arith.subi %[[TMP7]], %[[C1_3]] : i32
48! CHECK:    fir.store %[[TMP8]] to %[[TRIP]] : !fir.ref<i32>
49! CHECK:    %[[TMP9:.*]] = fir.load %[[I]] : !fir.ref<i32>
50! CHECK:    %[[C1_4:.*]] = arith.constant 1 : i32
51! CHECK:    %[[TMP10:.*]] = arith.addi %[[TMP9]], %[[C1_4]] overflow<nsw> : i32
52! CHECK:    fir.store %[[TMP10]] to %[[I]] : !fir.ref<i32>
53! CHECK:    cf.br ^[[HEADER]]
54  end do
55end subroutine
56! CHECK: ^[[EXIT]]:
57! CHECK:    return
58! CHECK:  }
59
60! Test jumping to the body of a do loop with a step expression.
61subroutine sub2()
62! CHECK-LABEL:  func @_QPsub2() {
63  implicit none
64  integer :: i, j
65  external foo
66! CHECK:    %[[TRIP:.*]] = fir.alloca i32
67! CHECK:    %[[I:.*]] = fir.alloca i32 {bindc_name = "i", {{.*}}}
68! CHECK:    %[[J:.*]] = fir.alloca i32 {bindc_name = "j", {{.*}}}
69
70  do i = 1, 3
71    if (i .eq. 2) goto 70
72! CHECK:    %[[TMP1:.*]] = fir.load %[[I]] : !fir.ref<i32>
73! CHECK:    %[[C2_1:.*]] = arith.constant 2 : i32
74! CHECK:    %[[COND:.*]] = arith.cmpi eq, %[[TMP1]], %[[C2_1]] : i32
75! CHECK:    cf.cond_br %[[COND]], ^[[COND_TRUE:.*]], ^{{.*}}
76! CHECK:  ^[[COND_TRUE]]:
77! CHECK:    cf.br ^[[BODY:.*]]
78
79  end do
80
81  call foo
82! CHECK:    fir.call @_QPfoo()
83
84  j = 3
85! CHECK:    %[[C3_1:.*]] = arith.constant 3 : i32
86! CHECK:    fir.store %[[C3_1]] to %[[J]] : !fir.ref<i32>
87
88  do i = 1, 2, 3 * j - 8
89! CHECK:    %[[C1_1:.*]] = arith.constant 1 : i32
90! CHECK:    %[[C2_2:.*]] = arith.constant 2 : i32
91! CHECK:    %[[C3_2:.*]] = arith.constant 3 : i32
92! CHECK:    %[[TMP2:.*]] = fir.load %[[J]] : !fir.ref<i32>
93! CHECK:    %[[TMP3:.*]] = arith.muli %[[C3_2]], %[[TMP2]] overflow<nsw> : i32
94! CHECK:    %[[C8_1:.*]] = arith.constant 8 : i32
95! CHECK:    %[[STEP:.*]] = arith.subi %[[TMP3]], %[[C8_1]] overflow<nsw> : i32
96! CHECK:    fir.store %[[STEP]] to %[[STEP_VAR:.*]] : !fir.ref<i32>
97! CHECK:    %[[TMP4:.*]] = arith.subi %[[C2_2]], %[[C1_1]] : i32
98! CHECK:    %[[TMP5:.*]] = arith.addi %[[TMP4]], %[[STEP]] : i32
99! CHECK:    %[[TMP6:.*]] = arith.divsi %[[TMP5]], %[[STEP]] : i32
100! CHECK:    fir.store %[[TMP6]] to %[[TRIP]] : !fir.ref<i32>
101! CHECK:    fir.store %[[C1_1]] to %[[I]] : !fir.ref<i32>
102! CHECK:    cf.br ^[[HEADER:.*]]
103! CHECK:  ^[[HEADER]]:
104! CHECK:    %[[TMP7:.*]] = fir.load %[[TRIP]] : !fir.ref<i32>
105! CHECK:    %[[C0_1:.*]] = arith.constant 0 : i32
106! CHECK:    %[[TMP8:.*]] = arith.cmpi sgt, %[[TMP7]], %[[C0_1]] : i32
107! CHECK:    cf.cond_br %[[TMP8]], ^[[BODY]], ^[[EXIT:.*]]
108
109    70 call foo
110! CHECK:  ^[[BODY]]:
111! CHECK:    fir.call @_QPfoo()
112! CHECK:    %[[TMP9:.*]] = fir.load %[[TRIP]] : !fir.ref<i32>
113! CHECK:    %[[C1_2:.*]] = arith.constant 1 : i32
114! CHECK:    %[[TMP10:.*]] = arith.subi %[[TMP9]], %[[C1_2]] : i32
115! CHECK:    fir.store %[[TMP10]] to %[[TRIP]] : !fir.ref<i32>
116! CHECK:    %[[TMP11:.*]] = fir.load %[[I]] : !fir.ref<i32>
117! CHECK:    %[[STEP_VAL:.*]] = fir.load %[[STEP_VAR]] : !fir.ref<i32>
118! CHECK:    %[[TMP12:.*]] = arith.addi %[[TMP11]], %[[STEP_VAL]] overflow<nsw> : i32
119! CHECK:    fir.store %[[TMP12]] to %[[I]] : !fir.ref<i32>
120! CHECK:    cf.br ^[[HEADER]]
121  end do
122end subroutine
123! CHECK: ^[[EXIT]]:
124! CHECK:    return
125! CHECK:  }
126