xref: /llvm-project/offload/test/offloading/fortran/target-map-pointer-scopes-enter-exit.f90 (revision 372344995568cae076477a8b0e98fcdec7c49379)

! Offloading test checking interaction of pointers with target in different
! scopes
! REQUIRES: flang, amdgpu

! RUN: %libomptarget-compile-fortran-run-and-check-generic
module test
    contains
  subroutine func_arg(arg_alloc)
    integer,  pointer, intent (inout) :: arg_alloc(:)

  !$omp target enter data map(alloc: arg_alloc)

  !$omp target
    do index = 1, 10
      arg_alloc(index) = arg_alloc(index) + index
    end do
  !$omp end target

  !$omp target exit data map(from: arg_alloc)

  !$omp target exit data map(delete: arg_alloc)

    print *, arg_alloc
  end subroutine func_arg
end module

subroutine func
  integer,  pointer :: local_alloc(:)
  allocate(local_alloc(10))

  !$omp target enter data map(alloc: local_alloc)

  !$omp target
    do index = 1, 10
      local_alloc(index) = index
    end do
  !$omp end target

  !$omp target exit data map(from: local_alloc)

  !$omp target exit data map(delete: local_alloc)

  print *, local_alloc

  deallocate(local_alloc)
end subroutine func


program main
  use test
  integer,  pointer :: map_ptr(:)
  allocate(map_ptr(10))

  !$omp target enter data map(alloc: map_ptr)

  !$omp target
    do index = 1, 10
      map_ptr(index) = index
    end do
  !$omp end target

  !$omp target exit data map(from: map_ptr)

  !$omp target exit data map(delete: map_ptr)

  call func

  print *, map_ptr

  call func_arg(map_ptr)

  deallocate(map_ptr)
end program

! CHECK: 1 2 3 4 5 6 7 8 9 10
! CHECK: 1 2 3 4 5 6 7 8 9 10
! CHECK: 2 4 6 8 10 12 14 16 18 20