250 lines · plain
1! REQUIRES: openmp_runtime2! RUN: bbc -emit-hlfir %openmp_flags -o - %s 2>&1 | FileCheck %s3! RUN: %flang_fc1 -emit-hlfir %openmp_flags -o - %s 2>&1 | FileCheck %s4 5! CHECK-LABEL: omp.private6! CHECK-SAME: {type = private} @[[LAST_PRIVATE_I:.*]] : i327 8! CHECK-LABEL: omp.private9! CHECK-SAME: {type = private} @[[LAST_PRIVATE_X:.*]] : i3210 11! CHECK-LABEL: omp.private12! CHECK-SAME: {type = private} @[[QFOMP_TASKLOOP_NOGROUPEI_PRIVATE_I32:.*]] : i3213 14! CHECK-LABEL: omp.private15! CHECK-SAME: {type = private} @[[OMP_TASKLOOP_UNTIEDEI_PRIVATE_I32:.*]] : i3216 17! CHECK-LABEL: omp.private18! CHECK-SAME: {type = private} @[[QFTEST_PRIORITYEI_PRIVATE_I32:.*]] : i3219 20! CHECK-LABEL: omp.private21! CHECK-SAME: {type = private} @[[QFTEST_MERGEABLEEI_PRIVATE_I32:.*]] : i3222 23! CHECK-LABEL: omp.private24! CHECK-SAME: {type = private} @[[I_PRIVATE_IF_TEST1:.*]] : i3225 26! CHECK-LABEL: omp.private27! CHECK-SAME: {type = private} @[[I_PRIVATE_FINAL:.*]] : i3228 29! CHECK-LABEL: omp.private30! CHECK-SAME: {type = private} @[[I_PRIVATE_TEST_ALLOCATE:.*]] : i3231 32! CHECK-LABEL: omp.private33! CHECK-SAME: {type = private} @[[X_PRIVATE_TEST_ALLOCATE:.*]] : i3234 35! CHECK-LABEL: omp.private 36! CHECK-SAME: {type = private} @[[I_PRIVATE_TEST2:.*]] : i3237 38! CHECK-LABEL: omp.private 39! CHECK-SAME: {type = private} @[[RES_PRIVATE_TEST2:.*]] : i3240 41! CHECK-LABEL: omp.private 42! CHECK-SAME: {type = private} @[[I_PRIVATE:.*]] : i3243 44! CHECK-LABEL: omp.private 45! CHECK-SAME: {type = firstprivate} @[[RES_FIRSTPRIVATE:.*]] : i32 46! CHECK-SAME: copy {47! CHECK: hlfir.assign 48 49! CHECK-LABEL: func.func @_QPomp_taskloop50! CHECK: %[[ALLOCA_I:.*]] = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFomp_taskloopEi"}51! CHECK: %[[I_VAL:.*]]:2 = hlfir.declare %[[ALLOCA_I]] {uniq_name = "_QFomp_taskloopEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)52! CHECK: %[[ALLOCA_RES:.*]] = fir.alloca i32 {bindc_name = "res", uniq_name = "_QFomp_taskloopEres"}53! CHECK: %[[RES_VAL:.*]]:2 = hlfir.declare %[[ALLOCA_RES]] {uniq_name = "_QFomp_taskloopEres"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)54! CHECK: %[[C1_I32:.*]] = arith.constant 1 : i3255! CHECK: %[[C10_I32:.*]] = arith.constant 10 : i3256! CHECK: %[[C1_I32_0:.*]] = arith.constant 1 : i3257! CHECK: omp.taskloop private(@[[RES_FIRSTPRIVATE]] %[[RES_VAL]]#0 -> %[[PRIV_RES:.*]], @[[I_PRIVATE]] %[[I_VAL]]#0 -> %[[PRIV_I:.*]] : !fir.ref<i32>, !fir.ref<i32>) {58! CHECK: omp.loop_nest (%[[ARG2:.*]]) : i32 = (%[[C1_I32]]) to (%[[C10_I32]]) inclusive step (%[[C1_I32_0]]) {59! CHECK: %[[RES_DECL:.*]]:2 = hlfir.declare %[[PRIV_RES]] {uniq_name = "_QFomp_taskloopEres"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)60! CHECK: %[[I_DECL:.*]]:2 = hlfir.declare %[[PRIV_I]] {uniq_name = "_QFomp_taskloopEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)61! CHECK: hlfir.assign %[[ARG2]] to %[[I_DECL]]#0 : i32, !fir.ref<i32>62! CHECK: %[[LOAD_RES:.*]] = fir.load %[[RES_DECL]]#0 : !fir.ref<i32>63! CHECK: %[[C1_I32_1:.*]] = arith.constant 1 : i3264! CHECK: %[[OUT_VAL:.*]] = arith.addi %[[LOAD_RES]], %[[C1_I32_1]] : i3265! CHECK: hlfir.assign %[[OUT_VAL]] to %[[RES_DECL]]#0 : i32, !fir.ref<i32>66! CHECK: omp.yield67! CHECK: }68! CHECK: }69! CHECK: return70! CHECK: }71 72subroutine omp_taskloop73 integer :: res, i74 !$omp taskloop75 do i = 1, 1076 res = res + 177 end do78 !$omp end taskloop79end subroutine omp_taskloop80 81 82! CHECK-LABEL: func.func @_QPomp_taskloop_private83! CHECK: %[[ALLOCA_I:.*]] = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFomp_taskloop_privateEi"}84! CHECK: %[[DECL_I:.*]]:2 = hlfir.declare %[[ALLOCA_I]] {uniq_name = "_QFomp_taskloop_privateEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)85! CHECK: %[[ALLOCA_RES:.*]] = fir.alloca i32 {bindc_name = "res", uniq_name = "_QFomp_taskloop_privateEres"}86! CHECK: %[[DECL_RES:.*]]:2 = hlfir.declare %[[ALLOCA_RES]] {uniq_name = "_QFomp_taskloop_privateEres"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)87subroutine omp_taskloop_private88 integer :: res, i89! CHECK: omp.taskloop private(@[[RES_PRIVATE_TEST2]] %[[DECL_RES]]#0 -> %[[ARG0:.*]], @[[I_PRIVATE_TEST2]] %[[DECL_I]]#0 -> %[[ARG1:.*]] : !fir.ref<i32>, !fir.ref<i32>) {90! CHECK: omp.loop_nest (%{{.*}}) : i32 = (%{{.*}}) to (%{{.*}}) inclusive step (%{{.*}}) {91! CHECK: %[[VAL1:.*]]:2 = hlfir.declare %[[ARG0]] {uniq_name = "_QFomp_taskloop_privateEres"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)92 !$omp taskloop private(res)93 do i = 1, 1094! CHECK: %[[LOAD_RES:.*]] = fir.load %[[VAL1]]#0 : !fir.ref<i32>95! CHECK: %[[C1_I32_1:.*]] = arith.constant 1 : i3296! CHECK: %[[ADD_VAL:.*]] = arith.addi %[[LOAD_RES]], %[[C1_I32_1]] : i3297! CHECK: hlfir.assign %[[ADD_VAL]] to %[[VAL1]]#0 : i32, !fir.ref<i32>98 res = res + 199 end do100! CHECK: return101! CHECK: }102 !$omp end taskloop103end subroutine omp_taskloop_private104 105!===============================================================================106! `allocate` clause107!===============================================================================108 109! CHECK-LABEL: func.func @_QPtaskloop_allocate110! CHECK: %[[ALLOCA_I:.*]] = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFtaskloop_allocateEi"}111! CHECK: %[[DECL_I:.*]]:2 = hlfir.declare %[[ALLOCA_I]] {uniq_name = "_QFtaskloop_allocateEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)112! CHECK: %[[ALLOCA_X:.*]] = fir.alloca i32 {bindc_name = "x", uniq_name = "_QFtaskloop_allocateEx"}113! CHECK: %[[DECL_X:.*]]:2 = hlfir.declare %[[ALLOCA_X]] {uniq_name = "_QFtaskloop_allocateEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)114subroutine taskloop_allocate()115 use omp_lib116 integer :: x117 ! CHECK: omp.taskloop allocate(%{{.*}} : i64 -> %[[DECL_X]]#0 : !fir.ref<i32>) 118 ! CHECK-SAME: private(@[[X_PRIVATE_TEST_ALLOCATE]] %[[DECL_X]]#0 -> %[[ARG0:.*]], @[[I_PRIVATE_TEST_ALLOCATE]] %[[DECL_I]]#0 -> %[[ARG1:.*]] : !fir.ref<i32>, !fir.ref<i32>) {119 !$omp taskloop allocate(omp_high_bw_mem_alloc: x) private(x)120 do i = 1, 100121 ! CHECK: arith.addi122 x = x + 12123 ! CHECK: omp.yield124 end do125 !$omp end taskloop126end subroutine taskloop_allocate127 128!===============================================================================129! `final` clause130!===============================================================================131 132! CHECK-LABEL: func.func @_QPtaskloop_final133! CHECK: %[[ALLOCA_I:.*]] = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFtaskloop_finalEi"}134! CHECK: %[[DECL_I:.*]]:2 = hlfir.declare %[[ALLOCA_I]] {uniq_name = "_QFtaskloop_finalEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)135subroutine taskloop_final()136 ! CHECK: omp.taskloop final(%true) private(@[[I_PRIVATE_FINAL]] %[[DECL_I]]#0 -> %[[ARG0:.*]] : !fir.ref<i32>) {137 !$omp taskloop final(.true.)138 do i = 1, 100139 ! CHECK: fir.call @_QPfoo()140 call foo()141 end do142 !$omp end taskloop143end subroutine144 145!===============================================================================146! `if` clause147!===============================================================================148 149! CHECK-LABEL: func.func @_QPomp_taskloop_if150! CHECK: %[[DECL_BAR:.*]]:2 = hlfir.declare %[[ARG0:.*]] dummy_scope %{{.*}}151! CHECK: %[[ALLOCA_I:.*]] = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFomp_taskloop_ifEi"}152! CHECK: %[[DECL_I:.*]]:2 = hlfir.declare %[[ALLOCA_I]] {uniq_name = "_QFomp_taskloop_ifEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)153! CHECK: %[[LOAD_VAL:.*]] = fir.load %[[DECL_BAR]]#0 : !fir.ref<!fir.logical<4>>154! CHECK: %[[VAL_BAR:.*]] = fir.convert %[[LOAD_VAL]] : (!fir.logical<4>) -> i1155subroutine omp_taskloop_if(bar)156 logical, intent(inout) :: bar157 !CHECK: omp.taskloop if(%[[VAL_BAR]]) private(@[[I_PRIVATE_IF_TEST1]] %[[DECL_I]]#0 -> %[[ARG1:.*]] : !fir.ref<i32>) {158 !$omp taskloop if(bar)159 do i = 1, 10160 call foo()161 end do162 !$omp end taskloop163end subroutine omp_taskloop_if164 165!===============================================================================166! `mergeable` clause167!===============================================================================168 169! CHECK-LABEL: func.func @_QPtest_mergeable170subroutine test_mergeable171 ! CHECK: omp.taskloop mergeable172 !$omp taskloop mergeable173 do i = 1, 10174 end do175 !$omp end taskloop176end subroutine test_mergeable177 178!===============================================================================179! `priority` clause180!===============================================================================181 182! CHECK-LABEL: func.func @_QPtest_priority183! CHECK: %[[VAL1:.*]]:2 = hlfir.declare %[[ARG0:.*]] dummy_scope %{{.*}}184! CHECK: %[[LOAD_VAL:.*]] = fir.load %[[VAL1]]#0 : !fir.ref<i32>185subroutine test_priority(n)186 integer, intent(inout) :: n187 ! CHECK: omp.taskloop priority(%[[LOAD_VAL]] : i32)188 !$omp taskloop priority(n)189 do i = 1, 10190 end do191 !$omp end taskloop192end subroutine test_priority193 194!===============================================================================195! `untied` clause196!===============================================================================197 198! CHECK-LABEL: func.func @_QPomp_taskloop_untied199subroutine omp_taskloop_untied()200 ! CHECK: omp.taskloop untied201 !$omp taskloop untied202 do i = 1, 10203 call foo()204 end do205 !$omp end taskloop206end subroutine207 208!===============================================================================209! `nogroup` clause210!===============================================================================211 212subroutine omp_taskloop_nogroup()213 ! CHECK: omp.taskloop nogroup214 !$omp taskloop nogroup215 do i = 1, 10216 call foo()217 end do218 !$omp end taskloop219end subroutine220 221!===============================================================================222! `lastprivate` clause223!===============================================================================224 225! CHECK-LABEL: func.func @_QPomp_taskloop_lastprivate226! CHECK: %[[ALLOCA_I:.*]] = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFomp_taskloop_lastprivateEi"}227! CHECK: %[[DECL_I:.*]]:2 = hlfir.declare %[[ALLOCA_I]] {uniq_name = "_QFomp_taskloop_lastprivateEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)228! CHECK: %[[ALLOCA_X:.*]] = fir.alloca i32 {bindc_name = "x", uniq_name = "_QFomp_taskloop_lastprivateEx"}229! CHECK: %[[DECL_X:.*]]:2 = hlfir.declare %[[ALLOCA_X]] {uniq_name = "_QFomp_taskloop_lastprivateEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)230subroutine omp_taskloop_lastprivate()231 integer x232 x = 0233 ! CHECK: omp.taskloop private(@[[LAST_PRIVATE_X]] %[[DECL_X]]#0 -> %[[ARG0]], @[[LAST_PRIVATE_I]] %[[DECL_I]]#0 -> %[[ARG1]] : !fir.ref<i32>, !fir.ref<i32>) {234 !$omp taskloop lastprivate(x)235 do i = 1, 100236 ! CHECK: %[[DECL_ARG0:.*]]:2 = hlfir.declare %[[ARG0]] {uniq_name = "_QFomp_taskloop_lastprivateEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)237 ! CHECK: %[[LOAD_ARG0:.*]] = fir.load %[[DECL_ARG0]]#0 : !fir.ref<i32>238 ! CHECK: %[[RES_ADD:.*]] = arith.addi %[[LOAD_ARG0]], %{{.*}} : i32239 ! CHECK: hlfir.assign %[[RES_ADD]] to %[[DECL_ARG0]]#0 : i32, !fir.ref<i32>240 x = x + 1241 ! CHECK: %[[SELCT_RESULT:.*]] = arith.select %{{.*}}, %{{.*}}, %{{.*}} : i1242 ! CHECK: fir.if %[[SELCT_RESULT]] {243 ! CHECK: %[[LOADED_SUM:.*]] = fir.load %[[DECL_ARG0]]#0 : !fir.ref<i32>244 ! CHECK: hlfir.assign %[[LOADED_SUM]] to %[[DECL_X]]#0 : i32, !fir.ref<i32>245 ! CHECK: }246 ! CHECK: omp.yield247 end do248 !$omp end taskloop249end subroutine omp_taskloop_lastprivate250