xref: /llvm-project/flang/test/Lower/alternate-return.f90 (revision 2985d1c267de69384f7be419d9cb01955537d6f2)
1! RUN: bbc -emit-fir -o - %s | FileCheck %s
2
3! CHECK-LABEL: func @_QPss
4subroutine ss(n)
5  print*, n
6  ! CHECK: return{{$}}
7  return
8! CHECK-LABEL: func @_QPee
9entry ee(n,*)
10  print*, n
11  ! CHECK: return %{{.}} : index
12  return 1
13end
14
15! CHECK-LABEL: func @_QQmain
16  call ss(7)
17  call ee(2, *3)
18  print*, 'default'
193 print*, 3
20
21  print*, k(10,20)
22  print*, k(15,15)
23  print*, k(20,10)
24end
25
26! CHECK-LABEL: func @_QPk
27function k(n1, n2)
28  ! CHECK-NOT: ^bb
29  ! CHECK: [[selector:%[0-9]+]] = fir.call @_QPs
30  ! CHECK-NEXT: fir.select [[selector]] : index [1, ^[[block1:bb[0-9]+]], 2, ^[[block2:bb[0-9]+]], unit, ^[[blockunit:bb[0-9]+]]
31  call s(n1, *5, n2, *7)
32  ! CHECK: ^[[blockunit]]: // pred: ^bb0
33  k =  0; return;
34  ! CHECK: ^[[block1]]: // pred: ^bb0
355 k = -1; return;
36  ! CHECK: ^[[block2]]: // pred: ^bb0
377 k =  1; return
38end
39
40! CHECK-LABEL: func @_QPs
41subroutine s(n1, *, n2, *)
42  ! CHECK: [[retval:%[0-9]+]] = fir.alloca index {{{.*}}bindc_name = "s"}
43  ! CHECK-COUNT-3: fir.store {{.*}} to [[retval]] : !fir.ref<index>
44  if (n1 < n2) return 1
45  if (n1 > n2) return 2
46  ! CHECK: {{.*}} = fir.load [[retval]] : !fir.ref<index>
47  ! CHECK-NEXT: return {{.*}} : index
48  return
49end
50