95 lines · plain
1! Tests mapping of a `do concurrent` loop with multiple iteration ranges.2 3! RUN: split-file %s %t4 5! RUN: %flang_fc1 -emit-hlfir -fopenmp -fdo-concurrent-to-openmp=host %t/multi_range.f90 -o - \6! RUN: | FileCheck %s --check-prefixes=HOST,COMMON7 8! RUN: %flang_fc1 -emit-hlfir -fopenmp -fdo-concurrent-to-openmp=device %t/multi_range.f90 -o - \9! RUN: | FileCheck %s --check-prefixes=DEVICE,COMMON10 11!--- multi_range.f9012program main13 integer, parameter :: n = 2014 integer, parameter :: m = 4015 integer, parameter :: l = 6016 integer :: a(n, m, l)17 18 do concurrent(i=3:n, j=5:m, k=7:l)19 a(i,j,k) = i * j + k20 end do21end22 23! COMMON: func.func @_QQmain24 25! COMMON: %[[C3:.*]] = arith.constant 3 : i3226! COMMON: %[[LB_I:.*]] = fir.convert %[[C3]] : (i32) -> index27! COMMON: %[[C20:.*]] = arith.constant 20 : i3228! COMMON: %[[UB_I:.*]] = fir.convert %[[C20]] : (i32) -> index29! COMMON: %[[STEP_I:.*]] = arith.constant 1 : index30 31! COMMON: %[[C5:.*]] = arith.constant 5 : i3232! COMMON: %[[LB_J:.*]] = fir.convert %[[C5]] : (i32) -> index33! COMMON: %[[C40:.*]] = arith.constant 40 : i3234! COMMON: %[[UB_J:.*]] = fir.convert %[[C40]] : (i32) -> index35! COMMON: %[[STEP_J:.*]] = arith.constant 1 : index36 37! COMMON: %[[C7:.*]] = arith.constant 7 : i3238! COMMON: %[[LB_K:.*]] = fir.convert %[[C7]] : (i32) -> index39! COMMON: %[[C60:.*]] = arith.constant 60 : i3240! COMMON: %[[UB_K:.*]] = fir.convert %[[C60]] : (i32) -> index41! COMMON: %[[STEP_K:.*]] = arith.constant 1 : index42 43! DEVICE: omp.target host_eval(44! DEVICE-SAME: %[[LB_I]] -> %[[LB_I:[[:alnum:]]+]],45! DEVICE-SAME: %[[UB_I]] -> %[[UB_I:[[:alnum:]]+]],46! DEVICE-SAME: %[[STEP_I]] -> %[[STEP_I:[[:alnum:]]+]],47! DEVICE-SAME: %[[LB_J]] -> %[[LB_J:[[:alnum:]]+]],48! DEVICE-SAME: %[[UB_J]] -> %[[UB_J:[[:alnum:]]+]],49! DEVICE-SAME: %[[STEP_J]] -> %[[STEP_J:[[:alnum:]]+]],50! DEVICE-SAME: %[[LB_K]] -> %[[LB_K:[[:alnum:]]+]],51! DEVICE-SAME: %[[UB_K]] -> %[[UB_K:[[:alnum:]]+]],52! DEVICE-SAME: %[[STEP_K]] -> %[[STEP_K:[[:alnum:]]+]] :53! DEVICE-SAME: index, index, index, index, index, index, index, index, index)54 55! DEVICE: omp.teams56 57! HOST-NOT: omp.target58! HOST-NOT: omp.teams59 60! COMMON: omp.parallel {61 62! COMMON-NEXT: %[[ITER_VAR_I:.*]] = fir.alloca i32 {bindc_name = "i"}63! COMMON-NEXT: %[[BINDING_I:.*]]:2 = hlfir.declare %[[ITER_VAR_I]] {uniq_name = "_QFEi"}64 65! COMMON-NEXT: %[[ITER_VAR_J:.*]] = fir.alloca i32 {bindc_name = "j"}66! COMMON-NEXT: %[[BINDING_J:.*]]:2 = hlfir.declare %[[ITER_VAR_J]] {uniq_name = "_QFEj"}67 68! COMMON-NEXT: %[[ITER_VAR_K:.*]] = fir.alloca i32 {bindc_name = "k"}69! COMMON-NEXT: %[[BINDING_K:.*]]:2 = hlfir.declare %[[ITER_VAR_K]] {uniq_name = "_QFEk"}70 71! DEVICE: omp.distribute72 73! COMMON: omp.wsloop {74! COMMON-NEXT: omp.loop_nest75! COMMON-SAME: (%[[ARG0:[^[:space:]]+]], %[[ARG1:[^[:space:]]+]], %[[ARG2:[^[:space:]]+]])76! COMMON-SAME: : index = (%[[LB_I]], %[[LB_J]], %[[LB_K]])77! COMMON-SAME: to (%[[UB_I]], %[[UB_J]], %[[UB_K]]) inclusive78! COMMON-SAME: step (%[[STEP_I]], %[[STEP_J]], %[[STEP_K]]) {79 80! COMMON-NEXT: %[[IV_IDX_I:.*]] = fir.convert %[[ARG0]]81! COMMON-NEXT: fir.store %[[IV_IDX_I]] to %[[BINDING_I]]#082 83! COMMON-NEXT: %[[IV_IDX_J:.*]] = fir.convert %[[ARG1]]84! COMMON-NEXT: fir.store %[[IV_IDX_J]] to %[[BINDING_J]]#085 86! COMMON-NEXT: %[[IV_IDX_K:.*]] = fir.convert %[[ARG2]]87! COMMON-NEXT: fir.store %[[IV_IDX_K]] to %[[BINDING_K]]#088 89! COMMON: omp.yield90! COMMON-NEXT: }91! COMMON-NEXT: }92 93! HOST-NEXT: omp.terminator94! HOST-NEXT: }95