blob: 9c527861c87b5facba248563a2212969f82bfc9b (
plain)
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
|
! Offloading test checking lowering of arrays with dynamic extents.
! REQUIRES: flang, amdgpu
! RUN: %libomptarget-compile-fortran-run-and-check-generic
program test_openmp_mapper
implicit none
integer, parameter :: n = 1024
type :: mytype
integer :: data(n)
end type mytype
type :: mytype2
type(mytype) :: my_data
end type mytype2
! Declare custom mappers for the derived type `mytype`
!$omp declare mapper(my_mapper1 : mytype :: t) map(to: t%data(1 : n))
! Declare custom mappers for the derived type `mytype2`
!$omp declare mapper(my_mapper2 : mytype2 :: t) map(mapper(my_mapper1): t%my_data)
type(mytype2) :: obj
integer :: i, sum_host, sum_device
! Initialize the host data
do i = 1, n
obj%my_data%data(i) = 1
end do
! Compute the sum on the host for verification
sum_host = sum(obj%my_data%data)
! Offload computation to the device using the named mapper `my_mapper2`
sum_device = 0
!$omp target map(tofrom: sum_device) map(mapper(my_mapper2) : obj)
do i = 1, n
sum_device = sum_device + obj%my_data%data(i)
end do
!$omp end target
! Check results
print *, "Sum on host: ", sum_host
print *, "Sum on device: ", sum_device
if (sum_device == sum_host) then
print *, "Test passed!"
else
print *, "Test failed!"
end if
end program test_openmp_mapper
! CHECK: Test passed!
|
