flang/test/Lower/HLFIR/assignment-intrinsics.f90

   1 ! Test lowering of intrinsic assignments to HLFIR
   2 ! RUN: bbc -emit-fir -hlfir -o - %s 2>&1 | FileCheck %s
   3
   4 ! -----------------------------------------------------------------------------
   5 !     Test assignments with scalar variable LHS and RHS
   6 ! -----------------------------------------------------------------------------
   7
   8 subroutine scalar_int(x, y)
   9   integer :: x, y
  10   x = y
  11 end subroutine
  12 ! CHECK-LABEL: func.func @_QPscalar_int(
  13 ! CHECK:  %[[VAL_2:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFscalar_intEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  14 ! CHECK:  %[[VAL_3:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFscalar_intEy"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  15 ! CHECK:  hlfir.assign %[[VAL_3]]#0 to %[[VAL_2]]#0 : !fir.ref<i32>, !fir.ref<i32>
  16
  17 subroutine scalar_logical(x, y)
  18   logical :: x, y
  19   x = y
  20 end subroutine
  21 ! CHECK-LABEL: func.func @_QPscalar_logical(
  22 ! CHECK:  %[[VAL_2:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFscalar_logicalEx"} : (!fir.ref<!fir.logical<4>>) -> (!fir.ref<!fir.logical<4>>, !fir.ref<!fir.logical<4>>)
  23 ! CHECK:  %[[VAL_3:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFscalar_logicalEy"} : (!fir.ref<!fir.logical<4>>) -> (!fir.ref<!fir.logical<4>>, !fir.ref<!fir.logical<4>>)
  24 ! CHECK:  hlfir.assign %[[VAL_3]]#0 to %[[VAL_2]]#0 : !fir.ref<!fir.logical<4>>, !fir.ref<!fir.logical<4>>
  25
  26 subroutine scalar_real(x, y)
  27   real :: x, y
  28   x = y
  29 end subroutine
  30 ! CHECK-LABEL: func.func @_QPscalar_real(
  31 ! CHECK:  %[[VAL_2:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFscalar_realEx"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)
  32 ! CHECK:  %[[VAL_3:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFscalar_realEy"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)
  33 ! CHECK:  hlfir.assign %[[VAL_3]]#0 to %[[VAL_2]]#0 : !fir.ref<f32>, !fir.ref<f32>
  34
  35 subroutine scalar_complex(x, y)
  36   complex :: x, y
  37   x = y
  38 end subroutine
  39 ! CHECK-LABEL: func.func @_QPscalar_complex(
  40 ! CHECK:  %[[VAL_2:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFscalar_complexEx"} : (!fir.ref<!fir.complex<4>>) -> (!fir.ref<!fir.complex<4>>, !fir.ref<!fir.complex<4>>)
  41 ! CHECK:  %[[VAL_3:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFscalar_complexEy"} : (!fir.ref<!fir.complex<4>>) -> (!fir.ref<!fir.complex<4>>, !fir.ref<!fir.complex<4>>)
  42 ! CHECK:  hlfir.assign %[[VAL_3]]#0 to %[[VAL_2]]#0 : !fir.ref<!fir.complex<4>>, !fir.ref<!fir.complex<4>>
  43
  44 subroutine scalar_character(x, y)
  45   character(*) :: x, y
  46   x = y
  47 end subroutine
  48 ! CHECK-LABEL: func.func @_QPscalar_character(
  49 ! CHECK:  %[[VAL_3:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFscalar_characterEx"} : (!fir.ref<!fir.char<1,?>>, index) -> (!fir.boxchar<1>, !fir.ref<!fir.char<1,?>>)
  50 ! CHECK:  %[[VAL_5:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFscalar_characterEy"} : (!fir.ref<!fir.char<1,?>>, index) -> (!fir.boxchar<1>, !fir.ref<!fir.char<1,?>>)
  51 ! CHECK:  hlfir.assign %[[VAL_5]]#0 to %[[VAL_3]]#0 : !fir.boxchar<1>, !fir.boxchar<1>
  52
  53 ! -----------------------------------------------------------------------------
  54 !     Test assignments with scalar variable LHS and expression RHS
  55 ! -----------------------------------------------------------------------------
  56
  57 subroutine scalar_int_2(x)
  58   integer :: x
  59   x = 42
  60 end subroutine
  61 ! CHECK-LABEL: func.func @_QPscalar_int_2(
  62 ! CHECK:  %[[VAL_1:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFscalar_int_2Ex"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  63 ! CHECK:  %[[VAL_2:.*]] = arith.constant 42 : i32
  64 ! CHECK:  hlfir.assign %[[VAL_2]] to %[[VAL_1]]#0 : i32, !fir.ref<i32>
  65
  66 subroutine scalar_logical_2(x)
  67   logical :: x
  68   x = .true.
  69 end subroutine
  70 ! CHECK-LABEL: func.func @_QPscalar_logical_2(
  71 ! CHECK:  %[[VAL_1:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFscalar_logical_2Ex"} : (!fir.ref<!fir.logical<4>>) -> (!fir.ref<!fir.logical<4>>, !fir.ref<!fir.logical<4>>)
  72 ! CHECK:  %[[VAL_2:.*]] = arith.constant true
  73 ! CHECK:  hlfir.assign %[[VAL_2]] to %[[VAL_1]]#0 : i1, !fir.ref<!fir.logical<4>>
  74
  75 subroutine scalar_real_2(x)
  76   real :: x
  77   x = 3.14
  78 end subroutine
  79 ! CHECK-LABEL: func.func @_QPscalar_real_2(
  80 ! CHECK:  %[[VAL_1:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFscalar_real_2Ex"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)
  81 ! CHECK:  %[[VAL_2:.*]] = arith.constant 3.140000e+00 : f32
  82 ! CHECK:  hlfir.assign %[[VAL_2]] to %[[VAL_1]]#0 : f32, !fir.ref<f32>
  83
  84 subroutine scalar_complex_2(x)
  85   complex :: x
  86   x = (1., -1.)
  87 end subroutine
  88 ! CHECK-LABEL: func.func @_QPscalar_complex_2(
  89 ! CHECK:  %[[VAL_1:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFscalar_complex_2Ex"} : (!fir.ref<!fir.complex<4>>) -> (!fir.ref<!fir.complex<4>>, !fir.ref<!fir.complex<4>>)
  90 ! CHECK:  %[[VAL_2:.*]] = arith.constant 1.000000e+00 : f32
  91 ! CHECK:  %[[VAL_3:.*]] = arith.constant -1.000000e+00 : f32
  92 ! CHECK:  %[[VAL_4:.*]] = fir.undefined !fir.complex<4>
  93 ! CHECK:  %[[VAL_5:.*]] = fir.insert_value %[[VAL_4]], %[[VAL_2]], [0 : index] : (!fir.complex<4>, f32) -> !fir.complex<4>
  94 ! CHECK:  %[[VAL_6:.*]] = fir.insert_value %[[VAL_5]], %[[VAL_3]], [1 : index] : (!fir.complex<4>, f32) -> !fir.complex<4>
  95 ! CHECK:  hlfir.assign %[[VAL_6]] to %[[VAL_1]]#0 : !fir.complex<4>, !fir.ref<!fir.complex<4>>
  96
  97 subroutine scalar_character_2(x)
  98   character(*) :: x
  99   x = "hello"
 100 end subroutine
 101 ! CHECK-LABEL: func.func @_QPscalar_character_2(
 102 ! CHECK:  %[[VAL_2:.*]]:2 = hlfir.declare {{.*}} {uniq_name = "_QFscalar_character_2Ex"} : (!fir.ref<!fir.char<1,?>>, index) -> (!fir.boxchar<1>, !fir.ref<!fir.char<1,?>>)
 103 ! CHECK:  %[[VAL_5:.*]]:2 = hlfir.declare {{.*}} {fortran_attrs = #fir.var_attrs<parameter>, uniq_name = "_QQcl.68656C6C6F"} : (!fir.ref<!fir.char<1,5>>, index) -> (!fir.ref<!fir.char<1,5>>, !fir.ref<!fir.char<1,5>>)
 104 ! CHECK:  hlfir.assign %[[VAL_5]]#0 to %[[VAL_2]]#0 : !fir.ref<!fir.char<1,5>>, !fir.boxchar<1>
 105
 106 ! -----------------------------------------------------------------------------
 107 !     Test assignments with array variable LHS and RHS
 108 ! -----------------------------------------------------------------------------
 109
 110 subroutine array(x, y)
 111   integer :: x(:), y(100)
 112   x = y
 113 end subroutine
 114 ! CHECK-LABEL: func.func @_QParray(
 115 ! CHECK:  %[[VAL_2:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFarrayEx"} : (!fir.box<!fir.array<?xi32>>) -> (!fir.box<!fir.array<?xi32>>, !fir.box<!fir.array<?xi32>>)
 116 ! CHECK:  %[[VAL_5:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFarrayEy"} : (!fir.ref<!fir.array<100xi32>>, !fir.shape<1>) -> (!fir.ref<!fir.array<100xi32>>, !fir.ref<!fir.array<100xi32>>)
 117 ! CHECK:  hlfir.assign %[[VAL_5]]#0 to %[[VAL_2]]#0 : !fir.ref<!fir.array<100xi32>>, !fir.box<!fir.array<?xi32>>
 118
 119 subroutine array_lbs(x, y)
 120   logical :: x(2:21), y(3:22)
 121   x = y
 122 end subroutine
 123 ! CHECK-LABEL: func.func @_QParray_lbs(
 124 ! CHECK:  %[[VAL_5:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFarray_lbsEx"} : (!fir.ref<!fir.array<20x!fir.logical<4>>>, !fir.shapeshift<1>) -> (!fir.box<!fir.array<20x!fir.logical<4>>>, !fir.ref<!fir.array<20x!fir.logical<4>>>)
 125 ! CHECK:  %[[VAL_9:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFarray_lbsEy"} : (!fir.ref<!fir.array<20x!fir.logical<4>>>, !fir.shapeshift<1>) -> (!fir.box<!fir.array<20x!fir.logical<4>>>, !fir.ref<!fir.array<20x!fir.logical<4>>>)
 126 ! CHECK:  hlfir.assign %[[VAL_9]]#0 to %[[VAL_5]]#0 : !fir.box<!fir.array<20x!fir.logical<4>>>, !fir.box<!fir.array<20x!fir.logical<4>>>
 127
 128
 129 subroutine array_character(x, y)
 130   character(*) :: x(10), y(10)
 131   x = y
 132 end subroutine
 133 ! CHECK-LABEL: func.func @_QParray_character(
 134 ! CHECK:  %[[VAL_6:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFarray_characterEx"} : (!fir.ref<!fir.array<10x!fir.char<1,?>>>, !fir.shape<1>, index) -> (!fir.box<!fir.array<10x!fir.char<1,?>>>, !fir.ref<!fir.array<10x!fir.char<1,?>>>)
 135 ! CHECK:  %[[VAL_11:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFarray_characterEy"} : (!fir.ref<!fir.array<10x!fir.char<1,?>>>, !fir.shape<1>, index) -> (!fir.box<!fir.array<10x!fir.char<1,?>>>, !fir.ref<!fir.array<10x!fir.char<1,?>>>)
 136 ! CHECK:  hlfir.assign %[[VAL_11]]#0 to %[[VAL_6]]#0 : !fir.box<!fir.array<10x!fir.char<1,?>>>, !fir.box<!fir.array<10x!fir.char<1,?>>>
 137
 138 subroutine array_pointer(x, y)
 139   real, pointer :: x(:), y(:)
 140   x = y
 141 end subroutine
 142 ! CHECK-LABEL: func.func @_QParray_pointer(
 143 ! CHECK:  %[[VAL_1:.*]]:2 = hlfir.declare %{{.*}}Ex
 144 ! CHECK:  %[[VAL_2:.*]]:2 = hlfir.declare %{{.*}}Ey
 145 ! CHECK:  %[[VAL_3:.*]] = fir.load %[[VAL_2]]#0 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>>
 146 ! CHECK:  %[[VAL_4:.*]] = fir.load %[[VAL_1]]#0 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>>
 147 ! CHECK:  hlfir.assign %[[VAL_3]] to %[[VAL_4]] : !fir.box<!fir.ptr<!fir.array<?xf32>>>, !fir.box<!fir.ptr<!fir.array<?xf32>>>
 148
 149 ! -----------------------------------------------------------------------------
 150 !     Test assignments with array LHS and scalar RHS
 151 ! -----------------------------------------------------------------------------
 152
 153 subroutine array_scalar(x, y)
 154   integer :: x(100), y
 155   x = y
 156 end subroutine
 157 ! CHECK-LABEL: func.func @_QParray_scalar(
 158 ! CHECK:  %[[VAL_4:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFarray_scalarEx"} : (!fir.ref<!fir.array<100xi32>>, !fir.shape<1>) -> (!fir.ref<!fir.array<100xi32>>, !fir.ref<!fir.array<100xi32>>)
 159 ! CHECK:  %[[VAL_5:.*]]:2 = hlfir.declare {{.*}}  {uniq_name = "_QFarray_scalarEy"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
 160 ! CHECK:  hlfir.assign %[[VAL_5]]#0 to %[[VAL_4]]#0 : !fir.ref<i32>, !fir.ref<!fir.array<100xi32>>