062fce6f4d
This pass will mark functions called from TargetOp's and declare target functions as implicitly declare target by adding the MLIR declare target attribute directly to the function. This pass executes after the initial lowering of Fortran's PFT to MLIR (FIR/OMP+Arith etc.) and is one of a series of passes that aim to clean up the MLIR for offloading (seperate passes in different patches, one for early outlining, another for declare target function filtering). Reviewers: jsjodin, skatrak, kiaranchandramohan Differential Revision: https://reviews.llvm.org/D154247
73 lines
3.6 KiB
Fortran
73 lines
3.6 KiB
Fortran
!RUN: %flang_fc1 -emit-fir -fopenmp %s -o - | FileCheck %s
|
|
!RUN: %flang_fc1 -emit-fir -fopenmp -fopenmp-is-target-device %s -o - | FileCheck %s
|
|
|
|
module test_0
|
|
implicit none
|
|
|
|
!CHECK-DAG: fir.global @_QMtest_0Edata_int {omp.declare_target = #omp.declaretarget<device_type = (any), capture_clause = (link)>} : i32
|
|
INTEGER :: data_int = 10
|
|
!$omp declare target link(data_int)
|
|
|
|
!CHECK-DAG: fir.global @_QMtest_0Earray_1d({{.*}}) {omp.declare_target = #omp.declaretarget<device_type = (any), capture_clause = (link)>} : !fir.array<3xi32>
|
|
INTEGER :: array_1d(3) = (/1,2,3/)
|
|
!$omp declare target link(array_1d)
|
|
|
|
!CHECK-DAG: fir.global @_QMtest_0Earray_2d({{.*}}) {omp.declare_target = #omp.declaretarget<device_type = (any), capture_clause = (link)>} : !fir.array<2x2xi32>
|
|
INTEGER :: array_2d(2,2) = reshape((/1,2,3,4/), (/2,2/))
|
|
!$omp declare target link(array_2d)
|
|
|
|
!CHECK-DAG: fir.global @_QMtest_0Ept1 {omp.declare_target = #omp.declaretarget<device_type = (any), capture_clause = (link)>} : !fir.box<!fir.ptr<i32>>
|
|
INTEGER, POINTER :: pt1
|
|
!$omp declare target link(pt1)
|
|
|
|
!CHECK-DAG: fir.global @_QMtest_0Ept2_tar {omp.declare_target = #omp.declaretarget<device_type = (any), capture_clause = (link)>} target : i32
|
|
INTEGER, TARGET :: pt2_tar = 5
|
|
!$omp declare target link(pt2_tar)
|
|
|
|
!CHECK-DAG: fir.global @_QMtest_0Ept2 {omp.declare_target = #omp.declaretarget<device_type = (any), capture_clause = (link)>} : !fir.box<!fir.ptr<i32>>
|
|
INTEGER, POINTER :: pt2 => pt2_tar
|
|
!$omp declare target link(pt2)
|
|
|
|
!CHECK-DAG: fir.global @_QMtest_0Edata_int_to {omp.declare_target = #omp.declaretarget<device_type = (any), capture_clause = (to)>} : i32
|
|
INTEGER :: data_int_to = 5
|
|
!$omp declare target to(data_int_to)
|
|
|
|
!CHECK-DAG: fir.global @_QMtest_0Edata_int_clauseless {omp.declare_target = #omp.declaretarget<device_type = (any), capture_clause = (to)>} : i32
|
|
INTEGER :: data_int_clauseless = 1
|
|
!$omp declare target(data_int_clauseless)
|
|
|
|
!CHECK-DAG: fir.global @_QMtest_0Edata_extended_to_1 {omp.declare_target = #omp.declaretarget<device_type = (any), capture_clause = (to)>} : f32
|
|
!CHECK-DAG: fir.global @_QMtest_0Edata_extended_to_2 {omp.declare_target = #omp.declaretarget<device_type = (any), capture_clause = (to)>} : f32
|
|
REAL :: data_extended_to_1 = 2
|
|
REAL :: data_extended_to_2 = 3
|
|
!$omp declare target to(data_extended_to_1, data_extended_to_2)
|
|
|
|
!CHECK-DAG: fir.global @_QMtest_0Edata_extended_link_1 {omp.declare_target = #omp.declaretarget<device_type = (any), capture_clause = (link)>} : f32
|
|
!CHECK-DAG: fir.global @_QMtest_0Edata_extended_link_2 {omp.declare_target = #omp.declaretarget<device_type = (any), capture_clause = (link)>} : f32
|
|
REAL :: data_extended_link_1 = 2
|
|
REAL :: data_extended_link_2 = 3
|
|
!$omp declare target link(data_extended_link_1, data_extended_link_2)
|
|
|
|
contains
|
|
end module test_0
|
|
|
|
PROGRAM commons
|
|
!CHECK-DAG: fir.global @_QCnumbers {omp.declare_target = #omp.declaretarget<device_type = (any), capture_clause = (to)>} : tuple<f32, f32> {
|
|
REAL :: one = 1
|
|
REAL :: two = 2
|
|
COMMON /numbers/ one, two
|
|
!$omp declare target(/numbers/)
|
|
|
|
!CHECK-DAG: fir.global @_QCnumbers_link {omp.declare_target = #omp.declaretarget<device_type = (any), capture_clause = (link)>} : tuple<f32, f32> {
|
|
REAL :: one_link = 1
|
|
REAL :: two_link = 2
|
|
COMMON /numbers_link/ one_link, two_link
|
|
!$omp declare target link(/numbers_link/)
|
|
|
|
!CHECK-DAG: fir.global @_QCnumbers_to {omp.declare_target = #omp.declaretarget<device_type = (any), capture_clause = (to)>} : tuple<f32, f32> {
|
|
REAL :: one_to = 1
|
|
REAL :: two_to = 2
|
|
COMMON /numbers_to/ one_to, two_to
|
|
!$omp declare target to(/numbers_to/)
|
|
END
|