libgomp/testsuite/libgomp.fortran/target-is-initial-host-2.f90

   1 ! { dg-additional-options "-fno-builtin-omp_is_initial_device" }
   2
   3 ! Check whether 'omp_is_initial_device()' is NOT compile-time optimized. */
   4
   5 ! { dg-additional-options "-fdump-tree-gimple -fdump-tree-optimized" }
   6 ! { dg-additional-options -foffload-options=-fdump-tree-optimized { target { offload_target_nvptx || offload_target_amdgcn } } }
   7
   8 ! { dg-final { scan-tree-dump-times "omp_is_initial_device" 1 "gimple" } }
   9
  10 ! { dg-final { scan-tree-dump-times "omp_is_initial_device" 1 "optimized" } }
  11
  12 ! { dg-final { only_for_offload_target amdgcn-amdhsa scan-offload-tree-dump-times "omp_is_initial_device" 1 "optimized" { target offload_target_amdgcn } } }
  13 ! { dg-final { only_for_offload_target nvptx-none scan-offload-tree-dump-times "omp_is_initial_device" 1 "optimized" { target offload_target_nvptx } } }
  14
  15
  16 program main
  17   use omp_lib
  18   implicit none (type, external)
  19   integer :: dev_num, initial, dev
  20   logical :: is_initial
  21
  22   initial = omp_get_initial_device()
  23   do dev = omp_initial_device, omp_get_num_devices()
  24       dev_num = 99
  25       !$omp target map(from: is_initial, dev_num) device(dev)
  26         is_initial = omp_is_initial_device ()
  27         dev_num = omp_get_device_num ()
  28       !$omp end target
  29       if (dev == omp_initial_device .or. dev == initial) then
  30         if (dev_num /= initial .or. .not. is_initial) &
  31           stop 1
  32       else
  33         if (dev_num /= dev .or. is_initial) &
  34           stop 2
  35       end if
  36   end do
  37 end