flang/test/Lower/OpenMP/target_data.f90

   1 !RUN: %flang_fc1 -emit-fir -fopenmp %s -o - | FileCheck %s
   2
   3 !===============================================================================
   4 ! Target_Enter Simple
   5 !===============================================================================
   6
   7 !CHECK-LABEL: func.func @_QPomp_target_enter_simple() {
   8 subroutine omp_target_enter_simple
   9    integer :: a(1024)
  10    !CHECK: omp.target_enter_data   map((to -> {{.*}} : !fir.ref<!fir.array<1024xi32>>))
  11    !$omp target enter data map(to: a)
  12 end subroutine omp_target_enter_simple
  13
  14 !===============================================================================
  15 ! Target_Enter Map types
  16 !===============================================================================
  17
  18 !CHECK-LABEL: func.func @_QPomp_target_enter_mt() {
  19 subroutine omp_target_enter_mt
  20    integer :: a(1024)
  21    integer :: b(1024)
  22    integer :: c(1024)
  23    integer :: d(1024)
  24    !CHECK: omp.target_enter_data   map((to -> {{.*}} : !fir.ref<!fir.array<1024xi32>>), (to -> {{.*}} : !fir.ref<!fir.array<1024xi32>>), (always, alloc -> {{.*}} : !fir.ref<!fir.array<1024xi32>>), (to -> {{.*}} : !fir.ref<!fir.array<1024xi32>>))
  25    !$omp target enter data map(to: a, b) map(always, alloc: c) map(to: d)
  26 end subroutine omp_target_enter_mt
  27
  28 !===============================================================================
  29 ! `Nowait` clause
  30 !===============================================================================
  31
  32 !CHECK-LABEL: func.func @_QPomp_target_enter_nowait() {
  33 subroutine omp_target_enter_nowait
  34    integer :: a(1024)
  35    !CHECK: omp.target_enter_data   nowait map((to -> {{.*}} : !fir.ref<!fir.array<1024xi32>>))
  36    !$omp target enter data map(to: a) nowait
  37 end subroutine omp_target_enter_nowait
  38
  39 !===============================================================================
  40 ! `if` clause
  41 !===============================================================================
  42
  43 !CHECK-LABEL: func.func @_QPomp_target_enter_if() {
  44 subroutine omp_target_enter_if
  45    integer :: a(1024)
  46    integer :: i
  47    i = 5
  48    !CHECK: %[[VAL_3:.*]] = fir.load %[[VAL_1:.*]] : !fir.ref<i32>
  49    !CHECK: %[[VAL_4:.*]] = arith.constant 10 : i32
  50    !CHECK: %[[VAL_5:.*]] = arith.cmpi slt, %[[VAL_3]], %[[VAL_4]] : i32
  51    !CHECK: omp.target_enter_data   if(%[[VAL_5]] : i1) map((to -> {{.*}} : !fir.ref<!fir.array<1024xi32>>))
  52    !$omp target enter data if(i<10) map(to: a)
  53 end subroutine omp_target_enter_if
  54
  55 !===============================================================================
  56 ! `device` clause
  57 !===============================================================================
  58
  59 !CHECK-LABEL: func.func @_QPomp_target_enter_device() {
  60 subroutine omp_target_enter_device
  61    integer :: a(1024)
  62    !CHECK: %[[VAL_1:.*]] = arith.constant 2 : i32
  63    !CHECK: omp.target_enter_data   device(%[[VAL_1]] : i32) map((to -> {{.*}} : !fir.ref<!fir.array<1024xi32>>))
  64    !$omp target enter data map(to: a) device(2)
  65 end subroutine omp_target_enter_device
  66
  67 !===============================================================================
  68 ! Target_Exit Simple
  69 !===============================================================================
  70
  71 !CHECK-LABEL: func.func @_QPomp_target_exit_simple() {
  72 subroutine omp_target_exit_simple
  73    integer :: a(1024)
  74    !CHECK: omp.target_exit_data   map((from -> {{.*}} : !fir.ref<!fir.array<1024xi32>>))
  75    !$omp target exit data map(from: a)
  76 end subroutine omp_target_exit_simple
  77
  78 !===============================================================================
  79 ! Target_Exit Map types
  80 !===============================================================================
  81
  82 !CHECK-LABEL: func.func @_QPomp_target_exit_mt() {
  83 subroutine omp_target_exit_mt
  84    integer :: a(1024)
  85    integer :: b(1024)
  86    integer :: c(1024)
  87    integer :: d(1024)
  88    integer :: e(1024)
  89    !CHECK: omp.target_exit_data   map((from -> {{.*}} : !fir.ref<!fir.array<1024xi32>>), (from -> {{.*}} : !fir.ref<!fir.array<1024xi32>>), (release -> {{.*}} : !fir.ref<!fir.array<1024xi32>>), (always, delete -> {{.*}} : !fir.ref<!fir.array<1024xi32>>), (from -> {{.*}} : !fir.ref<!fir.array<1024xi32>>))
  90    !$omp target exit data map(from: a,b) map(release: c) map(always, delete: d) map(from: e)
  91 end subroutine omp_target_exit_mt
  92
  93 !===============================================================================
  94 ! `device` clause
  95 !===============================================================================
  96
  97 !CHECK-LABEL: func.func @_QPomp_target_exit_device() {
  98 subroutine omp_target_exit_device
  99    integer :: a(1024)
 100    integer :: d
 101    !CHECK: %[[VAL_2:.*]] = fir.load %[[VAL_1:.*]] : !fir.ref<i32>
 102    !CHECK: omp.target_exit_data   device(%[[VAL_2]] : i32) map((from -> {{.*}} : !fir.ref<!fir.array<1024xi32>>))
 103    !$omp target exit data map(from: a) device(d)
 104 end subroutine omp_target_exit_device
 105
 106 !===============================================================================
 107 ! Target_Data with region
 108 !===============================================================================
 109
 110 !CHECK-LABEL: func.func @_QPomp_target_data() {
 111 subroutine omp_target_data
 112    !CHECK: %[[VAL_0:.*]] = fir.alloca !fir.array<1024xi32> {bindc_name = "a", uniq_name = "_QFomp_target_dataEa"}
 113    integer :: a(1024)
 114    !CHECK: omp.target_data   map((tofrom -> %[[VAL_0]] : !fir.ref<!fir.array<1024xi32>>)) {
 115    !$omp target data map(tofrom: a)
 116       !CHECK: %[[VAL_1:.*]] = arith.constant 10 : i32
 117       !CHECK: %[[VAL_2:.*]] = arith.constant 1 : i64
 118       !CHECK: %[[VAL_3:.*]] = arith.constant 1 : i64
 119       !CHECK: %[[VAL_4:.*]] = arith.subi %[[VAL_2]], %[[VAL_3]] : i64
 120       !CHECK: %[[VAL_5:.*]] = fir.coordinate_of %[[VAL_0]], %[[VAL_4]] : (!fir.ref<!fir.array<1024xi32>>, i64) -> !fir.ref<i32>
 121       !CHECK: fir.store %[[VAL_1]] to %[[VAL_5]] : !fir.ref<i32>
 122       a(1) = 10
 123    !CHECK: omp.terminator
 124    !$omp end target data
 125    !CHECK: }
 126 end subroutine omp_target_data