libgomp/testsuite/libgomp.fortran/target-is-accessible-1.f90

   1 program main
   2   use omp_lib
   3   use iso_c_binding
   4   implicit none (external, type)
   5   integer :: d, id, n, shared_mem, i
   6   integer, target :: a(1:128)
   7   type(c_ptr) :: p
   8
   9   d = omp_get_default_device ()
  10   id = omp_get_initial_device ()
  11   n = omp_get_num_devices ()
  12
  13   if (d < 0 .or. d >= n) &
  14     d = id
  15
  16   if (omp_target_is_accessible (p, c_sizeof (d), n) /= 1) &
  17     stop 1
  18
  19   if (omp_target_is_accessible (p, c_sizeof (d), id) /= 1) &
  20     stop 2
  21
  22   if (omp_target_is_accessible (p, c_sizeof (d), omp_initial_device) /= 1) &
  23     stop 3
  24
  25   if (omp_target_is_accessible (p, c_sizeof (d), -5) /= 0) &
  26     stop 4
  27
  28   if (omp_target_is_accessible (p, c_sizeof (d), n + 1) /= 0) &
  29     stop 5
  30
  31   ! Currently, a host pointer is accessible if the device supports shared
  32   ! memory or omp_target_is_accessible is executed on the host. This
  33   ! test case must be adapted when unified shared memory is avialable.
  34   do d = 0, omp_get_num_devices ()
  35     shared_mem = 0;
  36     !$omp target map (alloc: shared_mem) device (d)
  37       shared_mem = 1;
  38     !$omp end target
  39
  40     if (omp_target_is_accessible (p, c_sizeof (d), d) /= shared_mem) &
  41       stop 6;
  42
  43     if (omp_target_is_accessible (c_loc (a), 128 * sizeof (a(1)), d) /= shared_mem) &
  44       stop 7;
  45
  46     do i = 1, 128
  47       if (omp_target_is_accessible (c_loc (a(i)), sizeof (a(i)), d) /= shared_mem) &
  48         stop 8;
  49     end do
  50
  51   end do
  52
  53 end program main