6bfa042a10
This PR adds additional checks and tests for linear clause on omp.wsloop and omp.simd (both standalone and composite). For composite simd constructs, the translation to LLVMIR uses the same `LinearClauseProcessor` under `convertOmpSimd`, as already present in previous PRs like https://github.com/llvm/llvm-project/pull/150386 and https://github.com/llvm/llvm-project/pull/139386
109 lines
4.8 KiB
Fortran
109 lines
4.8 KiB
Fortran
! RUN: %flang_fc1 -fopenmp -emit-hlfir %s -o - 2>&1 | FileCheck %s
|
|
|
|
|
|
subroutine do_simd
|
|
!CHECK: %[[I:.*]]:2 = hlfir.declare %{{.*}} {uniq_name = "_QFdo_simdEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
|
|
!CHECK: %[[X:.*]]:2 = hlfir.declare %{{.*}} {uniq_name = "_QFdo_simdEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
|
|
!CHECK: %[[CONST:.*]] = arith.constant 1 : i32
|
|
!CHECK: %{{.*}} = arith.constant 1 : i32
|
|
!CHECK: %[[IV_STEP:.*]] = arith.constant 1 : i32
|
|
!CHECK: omp.wsloop {
|
|
!CHECK: omp.simd linear(%[[X]]#0 = %[[CONST]] : !fir.ref<i32>, %[[I]]#0 = %[[IV_STEP]] : !fir.ref<i32>) private(@_QFdo_simdEi_private_i32 {{.*}} -> %arg0 : !fir.ref<i32>) {
|
|
!CHECK: }
|
|
!CHECK: } {linear_var_types = [i32, i32], omp.composite}
|
|
!CHECK: } {omp.composite}
|
|
integer :: x
|
|
!$omp do simd linear(x:1)
|
|
do i = 1, N
|
|
end do
|
|
!$omp end do simd
|
|
end subroutine do_simd
|
|
|
|
|
|
subroutine distribute_simd
|
|
!CHECK: omp.teams {
|
|
!CHECK: omp.distribute private(@_QFdistribute_simdEi_private_i32 {{.*}} -> %[[ARG0:.*]] : !fir.ref<i32>) {
|
|
!CHECK: omp.simd linear(%[[ARG0]] = %c1_i32 : !fir.ref<i32>) private(@_QFdistribute_simdEi_private_i32 %[[ARG0]] -> {{.*}} : !fir.ref<i32>) {
|
|
!CHECK: } {linear_var_types = [i32], omp.composite}
|
|
!CHECK: } {omp.composite}
|
|
integer :: i
|
|
integer :: x
|
|
!$omp teams
|
|
!$omp distribute simd linear(i:1)
|
|
do i = 1, N
|
|
end do
|
|
!$omp end distribute simd
|
|
!$omp end teams
|
|
end subroutine distribute_simd
|
|
|
|
|
|
subroutine distribute_parallel_do
|
|
!CHECK: %[[I:.*]]:2 = hlfir.declare {{.*}} {uniq_name = "_QFdistribute_parallel_doEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
|
|
!CHECK: omp.teams {
|
|
!CHECK: omp.parallel {
|
|
!CHECK: %[[CONST]] = arith.constant 1 : i32
|
|
!CHECK: omp.distribute {
|
|
!CHECK: omp.wsloop {
|
|
!CHECK: omp.simd linear(%[[I]]#0 = %[[CONST]] : !fir.ref<i32>) private(@_QFdistribute_parallel_doEi_private_i32 %[[I]]#0 -> %arg0 : !fir.ref<i32>) {
|
|
!$omp teams
|
|
!$omp distribute parallel do simd linear(i:1)
|
|
do i = 1, N
|
|
end do
|
|
!$omp end distribute parallel do simd
|
|
!CHECK: } {linear_var_types = [i32], omp.composite}
|
|
!$omp end teams
|
|
end subroutine distribute_parallel_do
|
|
|
|
subroutine parallel_do
|
|
!CHECK: %[[I:.*]]:2 = hlfir.declare %{{.*}} {uniq_name = "_QFparallel_doEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
|
|
!CHECK: %[[X:.*]]:2 = hlfir.declare %{{.*}} {uniq_name = "_QFparallel_doEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
|
|
!CHECK: omp.parallel {
|
|
!CHECK: %[[LINEAR_STEP:.*]] = arith.constant 2 : i32
|
|
!CHECK: %{{.*}} = arith.constant 1 : i32
|
|
!CHECK: %[[IV_STEP:.*]] = arith.constant 1 : i32
|
|
!CHECK: omp.wsloop {
|
|
!CHECK: omp.simd linear(%[[X]]#0 = %[[LINEAR_STEP]] : !fir.ref<i32>, %[[I]]#0 = %[[IV_STEP]] : !fir.ref<i32>) private(@_QFparallel_doEi_private_i32 %[[I]]#0 -> %arg0 : !fir.ref<i32>) {
|
|
integer :: x
|
|
!$omp parallel do simd linear(x:2)
|
|
do i = 1, N
|
|
end do
|
|
!$omp end parallel do simd
|
|
!CHECK: } {linear_var_types = [i32, i32], omp.composite}
|
|
end subroutine parallel_do
|
|
|
|
subroutine teams_distribute
|
|
!CHECK: %[[I:.*]]:2 = hlfir.declare %{{.*}} {uniq_name = "_QFteams_distributeEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
|
|
!CHECK: %[[X:.*]]:2 = hlfir.declare %{{.*}} {uniq_name = "_QFteams_distributeEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
|
|
!CHECK: omp.teams {
|
|
!CHECK: %[[LINEAR_STEP:.*]] = arith.constant 1 : i32
|
|
!CHECK: {{.*}} = arith.constant 1 : i32
|
|
!CHECK: %[[IV_STEP:.*]] = arith.constant 1 : i32
|
|
!CHECK: omp.distribute {
|
|
!CHECK: omp.simd linear(%[[X]]#0 = %[[LINEAR_STEP]] : !fir.ref<i32>, %[[I]]#0 = %[[IV_STEP]] : !fir.ref<i32>) private(@_QFteams_distributeEi_private_i32 %[[I]]#0 -> %arg0 : !fir.ref<i32>) {
|
|
integer :: x
|
|
!$omp teams distribute simd linear(x)
|
|
do i = 1, N
|
|
end do
|
|
!$omp end teams distribute simd
|
|
!CHECK: } {linear_var_types = [i32, i32], omp.composite}
|
|
end subroutine teams_distribute
|
|
|
|
subroutine teams_distribute_parallel_do
|
|
!CHECK: %[[I:.*]]:2 = hlfir.declare %{{.*}} {uniq_name = "_QFteams_distribute_parallel_doEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
|
|
!CHECK: %[[X:.*]]:2 = hlfir.declare %{{.*}} {uniq_name = "_QFteams_distribute_parallel_doEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
|
|
!CHECK: omp.teams {
|
|
!CHECK: omp.parallel {
|
|
!CHECK: %[[LINEAR_STEP:.*]] = arith.constant 1 : i32
|
|
!CHECK: %{{.*}} = arith.constant 1 : i32
|
|
!CHECK: %[[IV_STEP:.*]] = arith.constant 1 : i32
|
|
!CHECK: omp.distribute {
|
|
!CHECK: omp.wsloop {
|
|
!CHECK: omp.simd linear(%[[X]]#0 = %c1_i32 : !fir.ref<i32>, %[[I]]#0 = %c1_i32_1 : !fir.ref<i32>) private(@_QFteams_distribute_parallel_doEi_private_i32 %[[I]]#0 -> %arg0 : !fir.ref<i32>) {
|
|
integer :: x
|
|
!$omp teams distribute parallel do simd linear(x)
|
|
do i = 1, N
|
|
end do
|
|
!$omp end teams distribute parallel do simd
|
|
!CHECK: } {linear_var_types = [i32, i32], omp.composite}
|
|
end subroutine teams_distribute_parallel_do
|