mirror of
https://github.com/OpenMP/Examples.git
synced 2025-04-04 05:41:33 +01:00
70 lines
1.7 KiB
Fortran
70 lines
1.7 KiB
Fortran
! @@name: declare_variant.1f90
|
|
! @@type: F-free
|
|
! @@compilable: yes, omp_5.0
|
|
! @@linkable: yes
|
|
! @@expect: success
|
|
|
|
module subs
|
|
use omp_lib
|
|
contains
|
|
subroutine vxv(v1, v2, v3) !! base function
|
|
integer,intent(in) :: v1(:),v2(:)
|
|
integer,intent(out) :: v3(:)
|
|
integer :: i,n
|
|
!$omp declare variant( p_vxv ) match( construct={parallel} )
|
|
!$omp declare variant( t_vxv ) match( construct={target} )
|
|
|
|
n=size(v1)
|
|
do i = 1,n; v3(i) = v1(i) * v2(i); enddo
|
|
|
|
end subroutine
|
|
|
|
subroutine p_vxv(v1, v2, v3) !! function variant
|
|
integer,intent(in) :: v1(:),v2(:)
|
|
integer,intent(out) :: v3(:)
|
|
integer :: i,n
|
|
n=size(v1)
|
|
|
|
!$omp do
|
|
do i = 1,n; v3(i) = v1(i) * v2(i) * 3; enddo
|
|
|
|
end subroutine
|
|
|
|
subroutine t_vxv(v1, v2, v3) !! function variant
|
|
integer,intent(in) :: v1(:),v2(:)
|
|
integer,intent(out) :: v3(:)
|
|
integer :: i,n
|
|
!$omp declare target
|
|
n=size(v1)
|
|
|
|
!$omp distribute simd
|
|
do i = 1,n; v3(i) = v1(i) * v2(i) * 2; enddo
|
|
|
|
end subroutine
|
|
|
|
end module subs
|
|
|
|
|
|
program main
|
|
use omp_lib
|
|
use subs
|
|
integer,parameter :: N = 100
|
|
integer :: v1(N), v2(N), v3(N)
|
|
|
|
do i= 1,N; v1(i)= i; v2(i)= -i; v3(i)= 0; enddo !! init
|
|
|
|
!$omp parallel
|
|
call vxv(v1,v2,v3)
|
|
!$omp end parallel
|
|
print *, v3(1),v3(N) !! from p_vxv -- output: -3 -30000
|
|
|
|
!$omp target teams map(to: v1,v2) map(from: v3)
|
|
call vxv(v1,v2,v3)
|
|
!$omp end target teams
|
|
print *, v3(1),v3(N) !! from t_vxv -- output: -2 -20000
|
|
|
|
call vxv(v1,v2,v3)
|
|
print *, v3(1),v3(N) !! from vxv -- output: -1 -10000
|
|
|
|
end program
|