You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
99 lines
3.2 KiB
Fortran
99 lines
3.2 KiB
Fortran
6 months ago
|
subroutine s_oacc_mlt_v_2(alpha, x, y, beta, z, info, conjgx, conjgy)
|
||
|
|
use psb_s_oacc_vect_mod, psb_protect_name => s_oacc_mlt_v_2
|
||
|
|
use psb_string_mod
|
||
|
|
implicit none
|
||
|
|
real(psb_spk_), intent(in) :: alpha, beta
|
||
|
|
class(psb_s_base_vect_type), intent(inout) :: x
|
||
|
|
class(psb_s_base_vect_type), intent(inout) :: y
|
||
|
|
class(psb_s_vect_oacc), intent(inout) :: z
|
||
|
|
integer(psb_ipk_), intent(out) :: info
|
||
|
|
character(len=1), intent(in), optional :: conjgx, conjgy
|
||
|
|
integer(psb_ipk_) :: i, n
|
||
|
|
logical :: conjgx_, conjgy_
|
||
|
|
|
||
|
|
conjgx_ = .false.
|
||
|
|
conjgy_ = .false.
|
||
|
|
if (present(conjgx)) conjgx_ = (psb_toupper(conjgx) == 'C')
|
||
|
|
if (present(conjgy)) conjgy_ = (psb_toupper(conjgy) == 'C')
|
||
|
|
|
||
|
|
n = min(x%get_nrows(), y%get_nrows(), z%get_nrows())
|
||
|
|
info = 0
|
||
|
|
select type(xx => x)
|
||
|
|
class is (psb_s_vect_oacc)
|
||
|
|
select type (yy => y)
|
||
|
|
class is (psb_s_vect_oacc)
|
||
|
|
if (xx%is_host()) call xx%sync()
|
||
|
|
if (yy%is_host()) call yy%sync()
|
||
|
|
if ((beta /= szero) .and. (z%is_host())) call z%sync()
|
||
|
|
if (conjgx_.and.conjgy_) then
|
||
|
|
!$acc parallel loop
|
||
|
|
do i = 1, n
|
||
|
|
z%v(i) = alpha * (xx%v(i)) * (yy%v(i)) + beta * z%v(i)
|
||
|
|
end do
|
||
|
|
else if (conjgx_.and.(.not.conjgy_)) then
|
||
|
|
!$acc parallel loop
|
||
|
|
do i = 1, n
|
||
|
|
z%v(i) = alpha * (xx%v(i)) * (yy%v(i)) + beta * z%v(i)
|
||
|
|
end do
|
||
|
|
else if ((.not.conjgx_).and.(conjgy_)) then
|
||
|
|
!$acc parallel loop
|
||
|
|
do i = 1, n
|
||
|
|
z%v(i) = alpha * (xx%v(i)) * (yy%v(i)) + beta * z%v(i)
|
||
|
|
end do
|
||
|
|
else
|
||
|
|
!$acc parallel loop
|
||
|
|
do i = 1, n
|
||
|
|
z%v(i) = alpha * (xx%v(i)) * (yy%v(i)) + beta * z%v(i)
|
||
|
|
end do
|
||
|
|
|
||
|
|
end if
|
||
|
|
call z%set_dev()
|
||
|
|
class default
|
||
|
|
if (xx%is_dev()) call xx%sync()
|
||
|
|
if (yy%is_dev()) call yy%sync()
|
||
|
|
if ((beta /= szero) .and. (z%is_dev())) call z%sync()
|
||
|
|
if (conjgx_.and.conjgy_) then
|
||
|
|
do i = 1, n
|
||
|
|
z%v(i) = alpha * (xx%v(i)) * (yy%v(i)) + beta * z%v(i)
|
||
|
|
end do
|
||
|
|
else if (conjgx_.and.(.not.conjgy_)) then
|
||
|
|
do i = 1, n
|
||
|
|
z%v(i) = alpha * (xx%v(i)) * (yy%v(i)) + beta * z%v(i)
|
||
|
|
end do
|
||
|
|
else if ((.not.conjgx_).and.(conjgy_)) then
|
||
|
|
do i = 1, n
|
||
|
|
z%v(i) = alpha * (xx%v(i)) * (yy%v(i)) + beta * z%v(i)
|
||
|
|
end do
|
||
|
|
else
|
||
|
|
do i = 1, n
|
||
|
|
z%v(i) = alpha * (xx%v(i)) * (yy%v(i)) + beta * z%v(i)
|
||
|
|
end do
|
||
|
|
end if
|
||
|
|
call z%set_host()
|
||
|
|
end select
|
||
|
|
class default
|
||
|
|
if (x%is_dev()) call x%sync()
|
||
|
|
if (y%is_dev()) call y%sync()
|
||
|
|
if ((beta /= szero) .and. (z%is_dev())) call z%sync()
|
||
|
|
if (conjgx_.and.conjgy_) then
|
||
|
|
do i = 1, n
|
||
|
|
z%v(i) = alpha * (x%v(i)) * (y%v(i)) + beta * z%v(i)
|
||
|
|
end do
|
||
|
|
else if (conjgx_.and.(.not.conjgy_)) then
|
||
|
|
do i = 1, n
|
||
|
|
z%v(i) = alpha * (x%v(i)) * (y%v(i)) + beta * z%v(i)
|
||
|
|
end do
|
||
|
|
else if ((.not.conjgx_).and.(conjgy_)) then
|
||
|
|
do i = 1, n
|
||
|
|
z%v(i) = alpha * (x%v(i)) * (y%v(i)) + beta * z%v(i)
|
||
|
|
end do
|
||
|
|
else
|
||
|
|
do i = 1, n
|
||
|
|
z%v(i) = alpha * (x%v(i)) * (y%v(i)) + beta * z%v(i)
|
||
|
|
end do
|
||
|
|
end if
|
||
|
|
call z%set_host()
|
||
|
|
end select
|
||
|
|
end subroutine s_oacc_mlt_v_2
|
||
|
|
|