224 lines · plain
1// RUN: fir-opt %s --split-input-file --pass-pipeline="builtin.module(test-acc-recipe-populate{recipe-type=private})" | FileCheck %s2 3// The tests here use a synthetic hlfir.declare in order to ensure that the hlfir dialect is4// loaded. This is required because the pass used is part of OpenACC test passes outside of5// flang and the APIs being test may generate hlfir even when it does not appear.6 7// Test scalar type (f32)8// CHECK: acc.private.recipe @private_scalar : !fir.ref<f32> init {9// CHECK: ^bb0(%{{.*}}: !fir.ref<f32>):10// CHECK: %[[ALLOC:.*]] = fir.alloca f3211// CHECK: %{{.*}}:2 = hlfir.declare %[[ALLOC]] {uniq_name = "scalar"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)12// CHECK: acc.yield %{{.*}}#0 : !fir.ref<f32>13// CHECK: }14// CHECK-NOT: destroy15 16func.func @test_scalar() {17 %0 = fir.alloca f32 {test.var = "scalar"}18 %var = fir.alloca f3219 %1:2 = hlfir.declare %var {uniq_name = "load_hlfir"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)20 return21}22 23// -----24 25// Test logical type26// CHECK: acc.private.recipe @private_logical : !fir.ref<!fir.logical<4>> init {27// CHECK: ^bb0(%{{.*}}: !fir.ref<!fir.logical<4>>):28// CHECK: %[[ALLOC:.*]] = fir.alloca !fir.logical<4>29// CHECK: %{{.*}}:2 = hlfir.declare %[[ALLOC]] {uniq_name = "logical"} : (!fir.ref<!fir.logical<4>>) -> (!fir.ref<!fir.logical<4>>, !fir.ref<!fir.logical<4>>)30// CHECK: acc.yield %{{.*}}#0 : !fir.ref<!fir.logical<4>>31// CHECK: }32// CHECK-NOT: destroy33 34func.func @test_logical() {35 %0 = fir.alloca !fir.logical<4> {test.var = "logical"}36 %var = fir.alloca f3237 %1:2 = hlfir.declare %var {uniq_name = "load_hlfir"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)38 return39}40 41// -----42 43// Test complex type44// CHECK: acc.private.recipe @private_complex : !fir.ref<complex<f32>> init {45// CHECK: ^bb0(%{{.*}}: !fir.ref<complex<f32>>):46// CHECK: %[[ALLOC:.*]] = fir.alloca complex<f32>47// CHECK: %{{.*}}:2 = hlfir.declare %[[ALLOC]] {uniq_name = "complex"} : (!fir.ref<complex<f32>>) -> (!fir.ref<complex<f32>>, !fir.ref<complex<f32>>)48// CHECK: acc.yield %{{.*}}#0 : !fir.ref<complex<f32>>49// CHECK: }50// CHECK-NOT: destroy51 52func.func @test_complex() {53 %0 = fir.alloca complex<f32> {test.var = "complex"}54 %var = fir.alloca f3255 %1:2 = hlfir.declare %var {uniq_name = "load_hlfir"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)56 return57}58 59// -----60 61// Test 1D static array62// CHECK: acc.private.recipe @private_array_1d : !fir.ref<!fir.array<100xf32>> init {63// CHECK: ^bb0(%{{.*}}: !fir.ref<!fir.array<100xf32>>):64// CHECK: %[[C100:.*]] = arith.constant 100 : index65// CHECK: %[[SHAPE:.*]] = fir.shape %[[C100]] : (index) -> !fir.shape<1>66// CHECK: %[[ALLOC:.*]] = fir.alloca !fir.array<100xf32>67// CHECK: %{{.*}}:2 = hlfir.declare %[[ALLOC]](%[[SHAPE]]) {uniq_name = "array_1d"} : (!fir.ref<!fir.array<100xf32>>, !fir.shape<1>) -> (!fir.ref<!fir.array<100xf32>>, !fir.ref<!fir.array<100xf32>>)68// CHECK: acc.yield %{{.*}}#0 : !fir.ref<!fir.array<100xf32>>69// CHECK: }70// CHECK-NOT: destroy71 72func.func @test_array_1d() {73 %0 = fir.alloca !fir.array<100xf32> {test.var = "array_1d"}74 %var = fir.alloca f3275 %1:2 = hlfir.declare %var {uniq_name = "load_hlfir"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)76 return77}78 79// -----80 81// Test 3D static array82// CHECK: acc.private.recipe @private_array_3d : !fir.ref<!fir.array<5x10x15xi32>> init {83// CHECK: ^bb0(%{{.*}}: !fir.ref<!fir.array<5x10x15xi32>>):84// CHECK: %[[C5:.*]] = arith.constant 5 : index85// CHECK: %[[C10:.*]] = arith.constant 10 : index86// CHECK: %[[C15:.*]] = arith.constant 15 : index87// CHECK: %[[SHAPE:.*]] = fir.shape %[[C5]], %[[C10]], %[[C15]] : (index, index, index) -> !fir.shape<3>88// CHECK: %[[ALLOC:.*]] = fir.alloca !fir.array<5x10x15xi32>89// CHECK: %{{.*}}:2 = hlfir.declare %[[ALLOC]](%[[SHAPE]]) {uniq_name = "array_3d"} : (!fir.ref<!fir.array<5x10x15xi32>>, !fir.shape<3>) -> (!fir.ref<!fir.array<5x10x15xi32>>, !fir.ref<!fir.array<5x10x15xi32>>)90// CHECK: acc.yield %{{.*}}#0 : !fir.ref<!fir.array<5x10x15xi32>>91// CHECK: }92// CHECK-NOT: destroy93 94func.func @test_array_3d() {95 %0 = fir.alloca !fir.array<5x10x15xi32> {test.var = "array_3d"}96 %var = fir.alloca f3297 %1:2 = hlfir.declare %var {uniq_name = "load_hlfir"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)98 return99}100 101// -----102 103// Test derived type with multiple fields104// CHECK: acc.private.recipe @private_derived : !fir.ref<!fir.type<_QTpoint{x:f32,y:f32,z:f32}>> init {105// CHECK: ^bb0(%{{.*}}: !fir.ref<!fir.type<_QTpoint{x:f32,y:f32,z:f32}>>):106// CHECK: %[[ALLOC:.*]] = fir.alloca !fir.type<_QTpoint{x:f32,y:f32,z:f32}>107// CHECK: %{{.*}}:2 = hlfir.declare %[[ALLOC]] {uniq_name = "derived"} : (!fir.ref<!fir.type<_QTpoint{x:f32,y:f32,z:f32}>>) -> (!fir.ref<!fir.type<_QTpoint{x:f32,y:f32,z:f32}>>, !fir.ref<!fir.type<_QTpoint{x:f32,y:f32,z:f32}>>)108// CHECK: acc.yield %{{.*}}#0 : !fir.ref<!fir.type<_QTpoint{x:f32,y:f32,z:f32}>>109// CHECK: }110// CHECK-NOT: destroy111 112func.func @test_derived() {113 %0 = fir.alloca !fir.type<_QTpoint{x:f32,y:f32,z:f32}> {test.var = "derived"}114 %var = fir.alloca f32115 %1:2 = hlfir.declare %var {uniq_name = "load_hlfir"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)116 return117}118 119// -----120 121// Test box type with heap scalar (needs destroy)122// CHECK: acc.private.recipe @private_box_heap_scalar : !fir.ref<!fir.box<!fir.heap<f64>>> init {123// CHECK: ^bb0(%{{.*}}: !fir.ref<!fir.box<!fir.heap<f64>>>):124// CHECK: %[[BOXALLOC:.*]] = fir.alloca !fir.box<!fir.heap<f64>>125// CHECK: %{{.*}}:2 = hlfir.declare %[[BOXALLOC]] {uniq_name = "box_heap_scalar"} : (!fir.ref<!fir.box<!fir.heap<f64>>>) -> (!fir.ref<!fir.box<!fir.heap<f64>>>, !fir.ref<!fir.box<!fir.heap<f64>>>)126// CHECK: %[[SCALAR:.*]] = fir.allocmem f64127// CHECK: %[[EMBOX:.*]] = fir.embox %[[SCALAR]] : (!fir.heap<f64>) -> !fir.box<!fir.heap<f64>>128// CHECK: fir.store %[[EMBOX]] to %{{.*}}#0 : !fir.ref<!fir.box<!fir.heap<f64>>>129// CHECK: acc.yield %{{.*}}#0 : !fir.ref<!fir.box<!fir.heap<f64>>>130// CHECK: } destroy {131// CHECK: ^bb0(%{{.*}}: !fir.ref<!fir.box<!fir.heap<f64>>>, %{{.*}}: !fir.ref<!fir.box<!fir.heap<f64>>>):132// CHECK: acc.terminator133// CHECK: }134 135func.func @test_box_heap_scalar() {136 %0 = fir.alloca !fir.box<!fir.heap<f64>> {test.var = "box_heap_scalar"}137 %var = fir.alloca f32138 %1:2 = hlfir.declare %var {uniq_name = "load_hlfir"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)139 return140}141 142// -----143 144// Test box type with pointer scalar (needs destroy)145// CHECK: acc.private.recipe @private_box_ptr_scalar : !fir.ref<!fir.box<!fir.ptr<i32>>> init {146// CHECK: ^bb0(%{{.*}}: !fir.ref<!fir.box<!fir.ptr<i32>>>):147// CHECK: %[[BOXALLOC:.*]] = fir.alloca !fir.box<!fir.ptr<i32>>148// CHECK: %{{.*}}:2 = hlfir.declare %[[BOXALLOC]] {uniq_name = "box_ptr_scalar"} : (!fir.ref<!fir.box<!fir.ptr<i32>>>) -> (!fir.ref<!fir.box<!fir.ptr<i32>>>, !fir.ref<!fir.box<!fir.ptr<i32>>>)149// CHECK: %[[SCALAR:.*]] = fir.allocmem i32150// CHECK: %[[EMBOX:.*]] = fir.embox %[[SCALAR]] : (!fir.heap<i32>) -> !fir.box<!fir.ptr<i32>>151// CHECK: fir.store %[[EMBOX]] to %{{.*}}#0 : !fir.ref<!fir.box<!fir.ptr<i32>>>152// CHECK: acc.yield %{{.*}}#0 : !fir.ref<!fir.box<!fir.ptr<i32>>>153// CHECK: } destroy {154// CHECK: ^bb0(%{{.*}}: !fir.ref<!fir.box<!fir.ptr<i32>>>, %{{.*}}: !fir.ref<!fir.box<!fir.ptr<i32>>>):155// CHECK: acc.terminator156// CHECK: }157 158func.func @test_box_ptr_scalar() {159 %0 = fir.alloca !fir.box<!fir.ptr<i32>> {test.var = "box_ptr_scalar"}160 %var = fir.alloca f32161 %1:2 = hlfir.declare %var {uniq_name = "load_hlfir"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)162 return163}164 165// -----166 167// Test box type with 1D heap array (needs destroy)168// CHECK: acc.private.recipe @private_box_heap_array_1d : !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>> init {169// CHECK: ^bb0(%{{.*}}: !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>):170// CHECK: %[[BOXALLOC:.*]] = fir.alloca !fir.box<!fir.heap<!fir.array<?xf32>>>171// CHECK: %{{.*}}:2 = hlfir.declare %[[BOXALLOC]] {uniq_name = "box_heap_array_1d"} : (!fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>) -> (!fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>, !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>)172// CHECK: acc.yield %{{.*}}#0 : !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>173// CHECK: } destroy {174// CHECK: ^bb0(%{{.*}}: !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>, %{{.*}}: !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>):175// CHECK: acc.terminator176// CHECK: }177 178func.func @test_box_heap_array_1d() {179 %0 = fir.alloca !fir.box<!fir.heap<!fir.array<?xf32>>> {test.var = "box_heap_array_1d"}180 %var = fir.alloca f32181 %1:2 = hlfir.declare %var {uniq_name = "load_hlfir"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)182 return183}184 185// -----186 187// Test box type with 2D heap array (needs destroy)188// CHECK: acc.private.recipe @private_box_heap_array_2d : !fir.ref<!fir.box<!fir.heap<!fir.array<?x?xi64>>>> init {189// CHECK: ^bb0(%{{.*}}: !fir.ref<!fir.box<!fir.heap<!fir.array<?x?xi64>>>>):190// CHECK: %[[BOXALLOC:.*]] = fir.alloca !fir.box<!fir.heap<!fir.array<?x?xi64>>>191// CHECK: %{{.*}}:2 = hlfir.declare %[[BOXALLOC]] {uniq_name = "box_heap_array_2d"} : (!fir.ref<!fir.box<!fir.heap<!fir.array<?x?xi64>>>>) -> (!fir.ref<!fir.box<!fir.heap<!fir.array<?x?xi64>>>>, !fir.ref<!fir.box<!fir.heap<!fir.array<?x?xi64>>>>)192// CHECK: acc.yield %{{.*}}#0 : !fir.ref<!fir.box<!fir.heap<!fir.array<?x?xi64>>>>193// CHECK: } destroy {194// CHECK: ^bb0(%{{.*}}: !fir.ref<!fir.box<!fir.heap<!fir.array<?x?xi64>>>>, %{{.*}}: !fir.ref<!fir.box<!fir.heap<!fir.array<?x?xi64>>>>):195// CHECK: acc.terminator196// CHECK: }197 198func.func @test_box_heap_array_2d() {199 %0 = fir.alloca !fir.box<!fir.heap<!fir.array<?x?xi64>>> {test.var = "box_heap_array_2d"}200 %var = fir.alloca f32201 %1:2 = hlfir.declare %var {uniq_name = "load_hlfir"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)202 return203}204 205// -----206 207// Test box type with pointer array (needs destroy)208// CHECK: acc.private.recipe @private_box_ptr_array : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>> init {209// CHECK: ^bb0(%{{.*}}: !fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>>):210// CHECK: %[[BOXALLOC:.*]] = fir.alloca !fir.box<!fir.ptr<!fir.array<?xf32>>>211// CHECK: %{{.*}}:2 = hlfir.declare %[[BOXALLOC]] {uniq_name = "box_ptr_array"} : (!fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>>) -> (!fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>>, !fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>>)212// CHECK: acc.yield %{{.*}}#0 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>>213// CHECK: } destroy {214// CHECK: ^bb0(%{{.*}}: !fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>>, %{{.*}}: !fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>>):215// CHECK: acc.terminator216// CHECK: }217 218func.func @test_box_ptr_array() {219 %0 = fir.alloca !fir.box<!fir.ptr<!fir.array<?xf32>>> {test.var = "box_ptr_array"}220 %var = fir.alloca f32221 %1:2 = hlfir.declare %var {uniq_name = "load_hlfir"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)222 return223}224