1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
! Offloading test which aims to test that an allocatable/descriptor type map
! will allow the appropriate slicing behaviour.
! REQUIRES: flang, amdgpu
subroutine slice_writer(n, a, b, c)
implicit none
integer, intent(in) :: n
real(8), intent(in) :: a(n)
real(8), intent(in) :: b(n)
real(8), intent(out) :: c(n)
integer :: i
!$omp target teams distribute parallel do
do i=1,n
c(i) = b(i) + a(i)
end do
end subroutine slice_writer
! RUN: %libomptarget-compile-fortran-run-and-check-generic
program main
implicit none
real(kind=8), allocatable :: a(:,:,:)
integer :: i, j, k, idx, idx1, idx2, idx3
i=50
j=100
k=2
allocate(a(1:i,1:j,1:k))
do idx1=1, i
do idx2=1, j
do idx3=1, k
a(idx1,idx2,idx3) = idx2
end do
end do
end do
do idx=1,k
!$omp target enter data map(alloc: a(1:i,:, idx))
!$omp target update to(a(1:i, 1:30, idx), &
!$omp& a(1:i, 61:100, idx))
call slice_writer(i, a(:, 1, idx), a(:, 61, idx), a(:, 31, idx))
call slice_writer(i, a(:, 30, idx), a(:, 100, idx), a(:, 60, idx))
!$omp target update from(a(1:i, 31:60, idx))
!$omp target exit data map(delete: a(1:i, :, idx))
print *, a(1, 31, idx), a(2, 31, idx), a(i, 31, idx)
print *, a(1, 60, idx), a(2, 60, idx), a(i, 60, idx)
enddo
deallocate(a)
end program
! CHECK: 62. 62. 62.
! CHECK: 130. 130. 130.
! CHECK: 62. 62. 62.
! CHECK: 130. 130. 130.
|
