fdurastante
/
psblas3
mirror of https://github.com/sfilippone/psblas3.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Initial version with OpenMP directives.

Browse Source
psblas-3.6-openmp
Salvatore Filippone 7 years ago
parent 00ac68739d
commit 8e7679dc02
17 changed files with 243 additions and 64 deletions
Show Stats Download Patch File Download Diff File

19
base/modules/serial/psb_c_base_vect_mod.f90
Unescape Escape View File

@ -964,6 +964,7 @@ contains
info = 0 info = 0
if (y%is_dev()) call y%sync() if (y%is_dev()) call y%sync()
n = min(size(y%v), size(x)) n = min(size(y%v), size(x))
!$omp parallel do private(i)
do i=1, n do i=1, n
y%v(i) = y%v(i)*x(i) y%v(i) = y%v(i)*x(i)
end do end do
@ -1001,6 +1002,7 @@ contains
if (beta == cone) then if (beta == cone) then
return return
else else
!$omp parallel do private(i) shared(beta)
do i=1, n do i=1, n
z%v(i) = beta*z%v(i) z%v(i) = beta*z%v(i)
end do end do
@ -1008,42 +1010,51 @@ contains
else else
if (alpha == cone) then if (alpha == cone) then
if (beta == czero) then if (beta == czero) then
!$omp parallel do private(i)
do i=1, n do i=1, n
z%v(i) = y(i)*x(i) z%v(i) = y(i)*x(i)
end do end do
else if (beta == cone) then else if (beta == cone) then
!$omp parallel do private(i)
do i=1, n do i=1, n
z%v(i) = z%v(i) + y(i)*x(i) z%v(i) = z%v(i) + y(i)*x(i)
end do end do
else else
!$omp parallel do private(i) shared(beta)
do i=1, n do i=1, n
z%v(i) = beta*z%v(i) + y(i)*x(i) z%v(i) = beta*z%v(i) + y(i)*x(i)
end do end do
end if end if
else if (alpha == -cone) then else if (alpha == -cone) then
if (beta == czero) then if (beta == czero) then
!$omp parallel do private(i)
do i=1, n do i=1, n
z%v(i) = -y(i)*x(i) z%v(i) = -y(i)*x(i)
end do end do
else if (beta == cone) then else if (beta == cone) then
!$omp parallel do private(i)
do i=1, n do i=1, n
z%v(i) = z%v(i) - y(i)*x(i) z%v(i) = z%v(i) - y(i)*x(i)
end do end do
else else
!$omp parallel do private(i) shared(beta)
do i=1, n do i=1, n
z%v(i) = beta*z%v(i) - y(i)*x(i) z%v(i) = beta*z%v(i) - y(i)*x(i)
end do end do
end if end if
else else
if (beta == czero) then if (beta == czero) then
!$omp parallel do private(i) shared(alpha)
do i=1, n do i=1, n
z%v(i) = alpha*y(i)*x(i) z%v(i) = alpha*y(i)*x(i)
end do end do
else if (beta == cone) then else if (beta == cone) then
!$omp parallel do private(i) shared(alpha)
do i=1, n do i=1, n
z%v(i) = z%v(i) + alpha*y(i)*x(i) z%v(i) = z%v(i) + alpha*y(i)*x(i)
end do end do
else else
!$omp parallel do private(i) shared(alpha,beta)
do i=1, n do i=1, n
z%v(i) = beta*z%v(i) + alpha*y(i)*x(i) z%v(i) = beta*z%v(i) + alpha*y(i)*x(i)
end do end do
@ -1144,7 +1155,9 @@ contains
complex(psb_spk_), intent (in) :: alpha complex(psb_spk_), intent (in) :: alpha
if (allocated(x%v)) then if (allocated(x%v)) then
!$OMP parallel workshare
x%v = alpha*x%v x%v = alpha*x%v
!$OMP end parallel workshare
call x%set_host() call x%set_host()
end if end if
@ -1181,8 +1194,9 @@ contains
real(psb_spk_) :: res real(psb_spk_) :: res
if (x%is_dev()) call x%sync() if (x%is_dev()) call x%sync()
!$OMP parallel workshare
res = maxval(abs(x%v(1:n))) res = maxval(abs(x%v(1:n)))
!$OMP end parallel workshare
end function c_base_amax end function c_base_amax
! !
@ -1197,8 +1211,9 @@ contains
real(psb_spk_) :: res real(psb_spk_) :: res
if (x%is_dev()) call x%sync() if (x%is_dev()) call x%sync()
!$OMP parallel workshare
res = sum(abs(x%v(1:n))) res = sum(abs(x%v(1:n)))
!$OMP end parallel workshare
end function c_base_asum end function c_base_asum

19
base/modules/serial/psb_d_base_vect_mod.f90
Unescape Escape View File

@ -964,6 +964,7 @@ contains
info = 0 info = 0
if (y%is_dev()) call y%sync() if (y%is_dev()) call y%sync()
n = min(size(y%v), size(x)) n = min(size(y%v), size(x))
!$omp parallel do private(i)
do i=1, n do i=1, n
y%v(i) = y%v(i)*x(i) y%v(i) = y%v(i)*x(i)
end do end do
@ -1001,6 +1002,7 @@ contains
if (beta == done) then if (beta == done) then
return return
else else
!$omp parallel do private(i) shared(beta)
do i=1, n do i=1, n
z%v(i) = beta*z%v(i) z%v(i) = beta*z%v(i)
end do end do
@ -1008,42 +1010,51 @@ contains
else else
if (alpha == done) then if (alpha == done) then
if (beta == dzero) then if (beta == dzero) then
!$omp parallel do private(i)
do i=1, n do i=1, n
z%v(i) = y(i)*x(i) z%v(i) = y(i)*x(i)
end do end do
else if (beta == done) then else if (beta == done) then
!$omp parallel do private(i)
do i=1, n do i=1, n
z%v(i) = z%v(i) + y(i)*x(i) z%v(i) = z%v(i) + y(i)*x(i)
end do end do
else else
!$omp parallel do private(i) shared(beta)
do i=1, n do i=1, n
z%v(i) = beta*z%v(i) + y(i)*x(i) z%v(i) = beta*z%v(i) + y(i)*x(i)
end do end do
end if end if
else if (alpha == -done) then else if (alpha == -done) then
if (beta == dzero) then if (beta == dzero) then
!$omp parallel do private(i)
do i=1, n do i=1, n
z%v(i) = -y(i)*x(i) z%v(i) = -y(i)*x(i)
end do end do
else if (beta == done) then else if (beta == done) then
!$omp parallel do private(i)
do i=1, n do i=1, n
z%v(i) = z%v(i) - y(i)*x(i) z%v(i) = z%v(i) - y(i)*x(i)
end do end do
else else
!$omp parallel do private(i) shared(beta)
do i=1, n do i=1, n
z%v(i) = beta*z%v(i) - y(i)*x(i) z%v(i) = beta*z%v(i) - y(i)*x(i)
end do end do
end if end if
else else
if (beta == dzero) then if (beta == dzero) then
!$omp parallel do private(i) shared(alpha)
do i=1, n do i=1, n
z%v(i) = alpha*y(i)*x(i) z%v(i) = alpha*y(i)*x(i)
end do end do
else if (beta == done) then else if (beta == done) then
!$omp parallel do private(i) shared(alpha)
do i=1, n do i=1, n
z%v(i) = z%v(i) + alpha*y(i)*x(i) z%v(i) = z%v(i) + alpha*y(i)*x(i)
end do end do
else else
!$omp parallel do private(i) shared(alpha,beta)
do i=1, n do i=1, n
z%v(i) = beta*z%v(i) + alpha*y(i)*x(i) z%v(i) = beta*z%v(i) + alpha*y(i)*x(i)
end do end do
@ -1144,7 +1155,9 @@ contains
real(psb_dpk_), intent (in) :: alpha real(psb_dpk_), intent (in) :: alpha
if (allocated(x%v)) then if (allocated(x%v)) then
!$OMP parallel workshare
x%v = alpha*x%v x%v = alpha*x%v
!$OMP end parallel workshare
call x%set_host() call x%set_host()
end if end if
@ -1181,8 +1194,9 @@ contains
real(psb_dpk_) :: res real(psb_dpk_) :: res
if (x%is_dev()) call x%sync() if (x%is_dev()) call x%sync()
!$OMP parallel workshare
res = maxval(abs(x%v(1:n))) res = maxval(abs(x%v(1:n)))
!$OMP end parallel workshare
end function d_base_amax end function d_base_amax
! !
@ -1197,8 +1211,9 @@ contains
real(psb_dpk_) :: res real(psb_dpk_) :: res
if (x%is_dev()) call x%sync() if (x%is_dev()) call x%sync()
!$OMP parallel workshare
res = sum(abs(x%v(1:n))) res = sum(abs(x%v(1:n)))
!$OMP end parallel workshare
end function d_base_asum end function d_base_asum

19
base/modules/serial/psb_s_base_vect_mod.f90
Unescape Escape View File

@ -964,6 +964,7 @@ contains
info = 0 info = 0
if (y%is_dev()) call y%sync() if (y%is_dev()) call y%sync()
n = min(size(y%v), size(x)) n = min(size(y%v), size(x))
!$omp parallel do private(i)
do i=1, n do i=1, n
y%v(i) = y%v(i)*x(i) y%v(i) = y%v(i)*x(i)
end do end do
@ -1001,6 +1002,7 @@ contains
if (beta == sone) then if (beta == sone) then
return return
else else
!$omp parallel do private(i) shared(beta)
do i=1, n do i=1, n
z%v(i) = beta*z%v(i) z%v(i) = beta*z%v(i)
end do end do
@ -1008,42 +1010,51 @@ contains
else else
if (alpha == sone) then if (alpha == sone) then
if (beta == szero) then if (beta == szero) then
!$omp parallel do private(i)
do i=1, n do i=1, n
z%v(i) = y(i)*x(i) z%v(i) = y(i)*x(i)
end do end do
else if (beta == sone) then else if (beta == sone) then
!$omp parallel do private(i)
do i=1, n do i=1, n
z%v(i) = z%v(i) + y(i)*x(i) z%v(i) = z%v(i) + y(i)*x(i)
end do end do
else else
!$omp parallel do private(i) shared(beta)
do i=1, n do i=1, n
z%v(i) = beta*z%v(i) + y(i)*x(i) z%v(i) = beta*z%v(i) + y(i)*x(i)
end do end do
end if end if
else if (alpha == -sone) then else if (alpha == -sone) then
if (beta == szero) then if (beta == szero) then
!$omp parallel do private(i)
do i=1, n do i=1, n
z%v(i) = -y(i)*x(i) z%v(i) = -y(i)*x(i)
end do end do
else if (beta == sone) then else if (beta == sone) then
!$omp parallel do private(i)
do i=1, n do i=1, n
z%v(i) = z%v(i) - y(i)*x(i) z%v(i) = z%v(i) - y(i)*x(i)
end do end do
else else
!$omp parallel do private(i) shared(beta)
do i=1, n do i=1, n
z%v(i) = beta*z%v(i) - y(i)*x(i) z%v(i) = beta*z%v(i) - y(i)*x(i)
end do end do
end if end if
else else
if (beta == szero) then if (beta == szero) then
!$omp parallel do private(i) shared(alpha)
do i=1, n do i=1, n
z%v(i) = alpha*y(i)*x(i) z%v(i) = alpha*y(i)*x(i)
end do end do
else if (beta == sone) then else if (beta == sone) then
!$omp parallel do private(i) shared(alpha)
do i=1, n do i=1, n
z%v(i) = z%v(i) + alpha*y(i)*x(i) z%v(i) = z%v(i) + alpha*y(i)*x(i)
end do end do
else else
!$omp parallel do private(i) shared(alpha,beta)
do i=1, n do i=1, n
z%v(i) = beta*z%v(i) + alpha*y(i)*x(i) z%v(i) = beta*z%v(i) + alpha*y(i)*x(i)
end do end do
@ -1144,7 +1155,9 @@ contains
real(psb_spk_), intent (in) :: alpha real(psb_spk_), intent (in) :: alpha
if (allocated(x%v)) then if (allocated(x%v)) then
!$OMP parallel workshare
x%v = alpha*x%v x%v = alpha*x%v
!$OMP end parallel workshare
call x%set_host() call x%set_host()
end if end if
@ -1181,8 +1194,9 @@ contains
real(psb_spk_) :: res real(psb_spk_) :: res
if (x%is_dev()) call x%sync() if (x%is_dev()) call x%sync()
!$OMP parallel workshare
res = maxval(abs(x%v(1:n))) res = maxval(abs(x%v(1:n)))
!$OMP end parallel workshare
end function s_base_amax end function s_base_amax
! !
@ -1197,8 +1211,9 @@ contains
real(psb_spk_) :: res real(psb_spk_) :: res
if (x%is_dev()) call x%sync() if (x%is_dev()) call x%sync()
!$OMP parallel workshare
res = sum(abs(x%v(1:n))) res = sum(abs(x%v(1:n)))
!$OMP end parallel workshare
end function s_base_asum end function s_base_asum

19
base/modules/serial/psb_z_base_vect_mod.f90
Unescape Escape View File

@ -964,6 +964,7 @@ contains
info = 0 info = 0
if (y%is_dev()) call y%sync() if (y%is_dev()) call y%sync()
n = min(size(y%v), size(x)) n = min(size(y%v), size(x))
!$omp parallel do private(i)
do i=1, n do i=1, n
y%v(i) = y%v(i)*x(i) y%v(i) = y%v(i)*x(i)
end do end do
@ -1001,6 +1002,7 @@ contains
if (beta == zone) then if (beta == zone) then
return return
else else
!$omp parallel do private(i) shared(beta)
do i=1, n do i=1, n
z%v(i) = beta*z%v(i) z%v(i) = beta*z%v(i)
end do end do
@ -1008,42 +1010,51 @@ contains
else else
if (alpha == zone) then if (alpha == zone) then
if (beta == zzero) then if (beta == zzero) then
!$omp parallel do private(i)
do i=1, n do i=1, n
z%v(i) = y(i)*x(i) z%v(i) = y(i)*x(i)
end do end do
else if (beta == zone) then else if (beta == zone) then
!$omp parallel do private(i)
do i=1, n do i=1, n
z%v(i) = z%v(i) + y(i)*x(i) z%v(i) = z%v(i) + y(i)*x(i)
end do end do
else else
!$omp parallel do private(i) shared(beta)
do i=1, n do i=1, n
z%v(i) = beta*z%v(i) + y(i)*x(i) z%v(i) = beta*z%v(i) + y(i)*x(i)
end do end do
end if end if
else if (alpha == -zone) then else if (alpha == -zone) then
if (beta == zzero) then if (beta == zzero) then
!$omp parallel do private(i)
do i=1, n do i=1, n
z%v(i) = -y(i)*x(i) z%v(i) = -y(i)*x(i)
end do end do
else if (beta == zone) then else if (beta == zone) then
!$omp parallel do private(i)
do i=1, n do i=1, n
z%v(i) = z%v(i) - y(i)*x(i) z%v(i) = z%v(i) - y(i)*x(i)
end do end do
else else
!$omp parallel do private(i) shared(beta)
do i=1, n do i=1, n
z%v(i) = beta*z%v(i) - y(i)*x(i) z%v(i) = beta*z%v(i) - y(i)*x(i)
end do end do
end if end if
else else
if (beta == zzero) then if (beta == zzero) then
!$omp parallel do private(i) shared(alpha)
do i=1, n do i=1, n
z%v(i) = alpha*y(i)*x(i) z%v(i) = alpha*y(i)*x(i)
end do end do
else if (beta == zone) then else if (beta == zone) then
!$omp parallel do private(i) shared(alpha)
do i=1, n do i=1, n
z%v(i) = z%v(i) + alpha*y(i)*x(i) z%v(i) = z%v(i) + alpha*y(i)*x(i)
end do end do
else else
!$omp parallel do private(i) shared(alpha,beta)
do i=1, n do i=1, n
z%v(i) = beta*z%v(i) + alpha*y(i)*x(i) z%v(i) = beta*z%v(i) + alpha*y(i)*x(i)
end do end do
@ -1144,7 +1155,9 @@ contains
complex(psb_dpk_), intent (in) :: alpha complex(psb_dpk_), intent (in) :: alpha
if (allocated(x%v)) then if (allocated(x%v)) then
!$OMP parallel workshare
x%v = alpha*x%v x%v = alpha*x%v
!$OMP end parallel workshare
call x%set_host() call x%set_host()
end if end if
@ -1181,8 +1194,9 @@ contains
real(psb_dpk_) :: res real(psb_dpk_) :: res
if (x%is_dev()) call x%sync() if (x%is_dev()) call x%sync()
!$OMP parallel workshare
res = maxval(abs(x%v(1:n))) res = maxval(abs(x%v(1:n)))
!$OMP end parallel workshare
end function z_base_amax end function z_base_amax
! !
@ -1197,8 +1211,9 @@ contains
real(psb_dpk_) :: res real(psb_dpk_) :: res
if (x%is_dev()) call x%sync() if (x%is_dev()) call x%sync()
!$OMP parallel workshare
res = sum(abs(x%v(1:n))) res = sum(abs(x%v(1:n)))
!$OMP end parallel workshare
end function z_base_asum end function z_base_asum

18
base/serial/impl/psb_c_csr_impl.f90
Unescape Escape View File

@ -131,10 +131,12 @@ contains
if (alpha == czero) then if (alpha == czero) then
if (beta == czero) then if (beta == czero) then
!$omp parallel do private(i)
do i = 1, m do i = 1, m
y(i) = czero y(i) = czero
enddo enddo
else else
!$omp parallel do private(i)
do i = 1, m do i = 1, m
y(i) = beta*y(i) y(i) = beta*y(i)
end do end do
@ -148,6 +150,7 @@ contains
if (beta == czero) then if (beta == czero) then
if (alpha == cone) then if (alpha == cone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = czero acc = czero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -158,6 +161,7 @@ contains
else if (alpha == -cone) then else if (alpha == -cone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = czero acc = czero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -168,6 +172,7 @@ contains
else else
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = czero acc = czero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -182,6 +187,7 @@ contains
else if (beta == cone) then else if (beta == cone) then
if (alpha == cone) then if (alpha == cone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = czero acc = czero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -192,6 +198,7 @@ contains
else if (alpha == -cone) then else if (alpha == -cone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = czero acc = czero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -202,6 +209,7 @@ contains
else else
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = czero acc = czero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -215,6 +223,7 @@ contains
else if (beta == -cone) then else if (beta == -cone) then
if (alpha == cone) then if (alpha == cone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = czero acc = czero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -224,7 +233,7 @@ contains
end do end do
else if (alpha == -cone) then else if (alpha == -cone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = czero acc = czero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -234,7 +243,7 @@ contains
end do end do
else else
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = czero acc = czero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -248,6 +257,7 @@ contains
else else
if (alpha == cone) then if (alpha == cone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = czero acc = czero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -257,7 +267,7 @@ contains
end do end do
else if (alpha == -cone) then else if (alpha == -cone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = czero acc = czero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -267,7 +277,7 @@ contains
end do end do
else else
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = czero acc = czero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1

18
base/serial/impl/psb_d_csr_impl.f90
Unescape Escape View File

@ -131,10 +131,12 @@ contains
if (alpha == dzero) then if (alpha == dzero) then
if (beta == dzero) then if (beta == dzero) then
!$omp parallel do private(i)
do i = 1, m do i = 1, m
y(i) = dzero y(i) = dzero
enddo enddo
else else
!$omp parallel do private(i)
do i = 1, m do i = 1, m
y(i) = beta*y(i) y(i) = beta*y(i)
end do end do
@ -148,6 +150,7 @@ contains
if (beta == dzero) then if (beta == dzero) then
if (alpha == done) then if (alpha == done) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = dzero acc = dzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -158,6 +161,7 @@ contains
else if (alpha == -done) then else if (alpha == -done) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = dzero acc = dzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -168,6 +172,7 @@ contains
else else
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = dzero acc = dzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -182,6 +187,7 @@ contains
else if (beta == done) then else if (beta == done) then
if (alpha == done) then if (alpha == done) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = dzero acc = dzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -192,6 +198,7 @@ contains
else if (alpha == -done) then else if (alpha == -done) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = dzero acc = dzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -202,6 +209,7 @@ contains
else else
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = dzero acc = dzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -215,6 +223,7 @@ contains
else if (beta == -done) then else if (beta == -done) then
if (alpha == done) then if (alpha == done) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = dzero acc = dzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -224,7 +233,7 @@ contains
end do end do
else if (alpha == -done) then else if (alpha == -done) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = dzero acc = dzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -234,7 +243,7 @@ contains
end do end do
else else
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = dzero acc = dzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -248,6 +257,7 @@ contains
else else
if (alpha == done) then if (alpha == done) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = dzero acc = dzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -257,7 +267,7 @@ contains
end do end do
else if (alpha == -done) then else if (alpha == -done) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = dzero acc = dzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -267,7 +277,7 @@ contains
end do end do
else else
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = dzero acc = dzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1

18
base/serial/impl/psb_s_csr_impl.f90
Unescape Escape View File

@ -131,10 +131,12 @@ contains
if (alpha == szero) then if (alpha == szero) then
if (beta == szero) then if (beta == szero) then
!$omp parallel do private(i)
do i = 1, m do i = 1, m
y(i) = szero y(i) = szero
enddo enddo
else else
!$omp parallel do private(i)
do i = 1, m do i = 1, m
y(i) = beta*y(i) y(i) = beta*y(i)
end do end do
@ -148,6 +150,7 @@ contains
if (beta == szero) then if (beta == szero) then
if (alpha == sone) then if (alpha == sone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = szero acc = szero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -158,6 +161,7 @@ contains
else if (alpha == -sone) then else if (alpha == -sone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = szero acc = szero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -168,6 +172,7 @@ contains
else else
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = szero acc = szero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -182,6 +187,7 @@ contains
else if (beta == sone) then else if (beta == sone) then
if (alpha == sone) then if (alpha == sone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = szero acc = szero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -192,6 +198,7 @@ contains
else if (alpha == -sone) then else if (alpha == -sone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = szero acc = szero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -202,6 +209,7 @@ contains
else else
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = szero acc = szero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -215,6 +223,7 @@ contains
else if (beta == -sone) then else if (beta == -sone) then
if (alpha == sone) then if (alpha == sone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = szero acc = szero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -224,7 +233,7 @@ contains
end do end do
else if (alpha == -sone) then else if (alpha == -sone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = szero acc = szero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -234,7 +243,7 @@ contains
end do end do
else else
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = szero acc = szero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -248,6 +257,7 @@ contains
else else
if (alpha == sone) then if (alpha == sone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = szero acc = szero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -257,7 +267,7 @@ contains
end do end do
else if (alpha == -sone) then else if (alpha == -sone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = szero acc = szero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -267,7 +277,7 @@ contains
end do end do
else else
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = szero acc = szero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1

18
base/serial/impl/psb_z_csr_impl.f90
Unescape Escape View File

@ -131,10 +131,12 @@ contains
if (alpha == zzero) then if (alpha == zzero) then
if (beta == zzero) then if (beta == zzero) then
!$omp parallel do private(i)
do i = 1, m do i = 1, m
y(i) = zzero y(i) = zzero
enddo enddo
else else
!$omp parallel do private(i)
do i = 1, m do i = 1, m
y(i) = beta*y(i) y(i) = beta*y(i)
end do end do
@ -148,6 +150,7 @@ contains
if (beta == zzero) then if (beta == zzero) then
if (alpha == zone) then if (alpha == zone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = zzero acc = zzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -158,6 +161,7 @@ contains
else if (alpha == -zone) then else if (alpha == -zone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = zzero acc = zzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -168,6 +172,7 @@ contains
else else
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = zzero acc = zzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -182,6 +187,7 @@ contains
else if (beta == zone) then else if (beta == zone) then
if (alpha == zone) then if (alpha == zone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = zzero acc = zzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -192,6 +198,7 @@ contains
else if (alpha == -zone) then else if (alpha == -zone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = zzero acc = zzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -202,6 +209,7 @@ contains
else else
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = zzero acc = zzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -215,6 +223,7 @@ contains
else if (beta == -zone) then else if (beta == -zone) then
if (alpha == zone) then if (alpha == zone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = zzero acc = zzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -224,7 +233,7 @@ contains
end do end do
else if (alpha == -zone) then else if (alpha == -zone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = zzero acc = zzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -234,7 +243,7 @@ contains
end do end do
else else
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = zzero acc = zzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -248,6 +257,7 @@ contains
else else
if (alpha == zone) then if (alpha == zone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = zzero acc = zzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -257,7 +267,7 @@ contains
end do end do
else if (alpha == -zone) then else if (alpha == -zone) then
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = zzero acc = zzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1
@ -267,7 +277,7 @@ contains
end do end do
else else
!$omp parallel do private(i,j,acc)
do i=1,m do i=1,m
acc = zzero acc = zzero
do j=irp(i), irp(i+1)-1 do j=irp(i), irp(i+1)-1

15
base/serial/psi_c_serial_impl.f90
Unescape Escape View File

@ -476,6 +476,7 @@ subroutine caxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (alpha.eq.czero) then if (alpha.eq.czero) then
if (beta.eq.czero) then if (beta.eq.czero) then
do j=1, n do j=1, n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = czero y(i,j) = czero
enddo enddo
@ -487,12 +488,14 @@ subroutine caxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-cone) then else if (beta.eq.-cone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = - y(i,j) y(i,j) = - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = beta*y(i,j) y(i,j) = beta*y(i,j)
enddo enddo
@ -503,12 +506,14 @@ subroutine caxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (beta.eq.czero) then if (beta.eq.czero) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) y(i,j) = x(i,j)
enddo enddo
enddo enddo
else if (beta.eq.cone) then else if (beta.eq.cone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) + y(i,j) y(i,j) = x(i,j) + y(i,j)
enddo enddo
@ -516,12 +521,14 @@ subroutine caxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-cone) then else if (beta.eq.-cone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) - y(i,j) y(i,j) = x(i,j) - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) + beta*y(i,j) y(i,j) = x(i,j) + beta*y(i,j)
enddo enddo
@ -532,12 +539,14 @@ subroutine caxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (beta.eq.czero) then if (beta.eq.czero) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) y(i,j) = -x(i,j)
enddo enddo
enddo enddo
else if (beta.eq.cone) then else if (beta.eq.cone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) + y(i,j) y(i,j) = -x(i,j) + y(i,j)
enddo enddo
@ -545,12 +554,14 @@ subroutine caxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-cone) then else if (beta.eq.-cone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) - y(i,j) y(i,j) = -x(i,j) - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) + beta*y(i,j) y(i,j) = -x(i,j) + beta*y(i,j)
enddo enddo
@ -561,12 +572,14 @@ subroutine caxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (beta.eq.czero) then if (beta.eq.czero) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) y(i,j) = alpha*x(i,j)
enddo enddo
enddo enddo
else if (beta.eq.cone) then else if (beta.eq.cone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) + y(i,j) y(i,j) = alpha*x(i,j) + y(i,j)
enddo enddo
@ -574,12 +587,14 @@ subroutine caxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-cone) then else if (beta.eq.-cone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) - y(i,j) y(i,j) = alpha*x(i,j) - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) + beta*y(i,j) y(i,j) = alpha*x(i,j) + beta*y(i,j)
enddo enddo

15
base/serial/psi_d_serial_impl.f90
Unescape Escape View File

@ -476,6 +476,7 @@ subroutine daxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (alpha.eq.dzero) then if (alpha.eq.dzero) then
if (beta.eq.dzero) then if (beta.eq.dzero) then
do j=1, n do j=1, n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = dzero y(i,j) = dzero
enddo enddo
@ -487,12 +488,14 @@ subroutine daxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-done) then else if (beta.eq.-done) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = - y(i,j) y(i,j) = - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = beta*y(i,j) y(i,j) = beta*y(i,j)
enddo enddo
@ -503,12 +506,14 @@ subroutine daxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (beta.eq.dzero) then if (beta.eq.dzero) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) y(i,j) = x(i,j)
enddo enddo
enddo enddo
else if (beta.eq.done) then else if (beta.eq.done) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) + y(i,j) y(i,j) = x(i,j) + y(i,j)
enddo enddo
@ -516,12 +521,14 @@ subroutine daxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-done) then else if (beta.eq.-done) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) - y(i,j) y(i,j) = x(i,j) - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) + beta*y(i,j) y(i,j) = x(i,j) + beta*y(i,j)
enddo enddo
@ -532,12 +539,14 @@ subroutine daxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (beta.eq.dzero) then if (beta.eq.dzero) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) y(i,j) = -x(i,j)
enddo enddo
enddo enddo
else if (beta.eq.done) then else if (beta.eq.done) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) + y(i,j) y(i,j) = -x(i,j) + y(i,j)
enddo enddo
@ -545,12 +554,14 @@ subroutine daxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-done) then else if (beta.eq.-done) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) - y(i,j) y(i,j) = -x(i,j) - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) + beta*y(i,j) y(i,j) = -x(i,j) + beta*y(i,j)
enddo enddo
@ -561,12 +572,14 @@ subroutine daxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (beta.eq.dzero) then if (beta.eq.dzero) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) y(i,j) = alpha*x(i,j)
enddo enddo
enddo enddo
else if (beta.eq.done) then else if (beta.eq.done) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) + y(i,j) y(i,j) = alpha*x(i,j) + y(i,j)
enddo enddo
@ -574,12 +587,14 @@ subroutine daxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-done) then else if (beta.eq.-done) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) - y(i,j) y(i,j) = alpha*x(i,j) - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) + beta*y(i,j) y(i,j) = alpha*x(i,j) + beta*y(i,j)
enddo enddo

15
base/serial/psi_i_serial_impl.f90
Unescape Escape View File

@ -476,6 +476,7 @@ subroutine iaxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (alpha.eq.izero) then if (alpha.eq.izero) then
if (beta.eq.izero) then if (beta.eq.izero) then
do j=1, n do j=1, n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = izero y(i,j) = izero
enddo enddo
@ -487,12 +488,14 @@ subroutine iaxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-ione) then else if (beta.eq.-ione) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = - y(i,j) y(i,j) = - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = beta*y(i,j) y(i,j) = beta*y(i,j)
enddo enddo
@ -503,12 +506,14 @@ subroutine iaxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (beta.eq.izero) then if (beta.eq.izero) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) y(i,j) = x(i,j)
enddo enddo
enddo enddo
else if (beta.eq.ione) then else if (beta.eq.ione) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) + y(i,j) y(i,j) = x(i,j) + y(i,j)
enddo enddo
@ -516,12 +521,14 @@ subroutine iaxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-ione) then else if (beta.eq.-ione) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) - y(i,j) y(i,j) = x(i,j) - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) + beta*y(i,j) y(i,j) = x(i,j) + beta*y(i,j)
enddo enddo
@ -532,12 +539,14 @@ subroutine iaxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (beta.eq.izero) then if (beta.eq.izero) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) y(i,j) = -x(i,j)
enddo enddo
enddo enddo
else if (beta.eq.ione) then else if (beta.eq.ione) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) + y(i,j) y(i,j) = -x(i,j) + y(i,j)
enddo enddo
@ -545,12 +554,14 @@ subroutine iaxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-ione) then else if (beta.eq.-ione) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) - y(i,j) y(i,j) = -x(i,j) - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) + beta*y(i,j) y(i,j) = -x(i,j) + beta*y(i,j)
enddo enddo
@ -561,12 +572,14 @@ subroutine iaxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (beta.eq.izero) then if (beta.eq.izero) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) y(i,j) = alpha*x(i,j)
enddo enddo
enddo enddo
else if (beta.eq.ione) then else if (beta.eq.ione) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) + y(i,j) y(i,j) = alpha*x(i,j) + y(i,j)
enddo enddo
@ -574,12 +587,14 @@ subroutine iaxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-ione) then else if (beta.eq.-ione) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) - y(i,j) y(i,j) = alpha*x(i,j) - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) + beta*y(i,j) y(i,j) = alpha*x(i,j) + beta*y(i,j)
enddo enddo

15
base/serial/psi_s_serial_impl.f90
Unescape Escape View File

@ -476,6 +476,7 @@ subroutine saxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (alpha.eq.szero) then if (alpha.eq.szero) then
if (beta.eq.szero) then if (beta.eq.szero) then
do j=1, n do j=1, n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = szero y(i,j) = szero
enddo enddo
@ -487,12 +488,14 @@ subroutine saxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-sone) then else if (beta.eq.-sone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = - y(i,j) y(i,j) = - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = beta*y(i,j) y(i,j) = beta*y(i,j)
enddo enddo
@ -503,12 +506,14 @@ subroutine saxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (beta.eq.szero) then if (beta.eq.szero) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) y(i,j) = x(i,j)
enddo enddo
enddo enddo
else if (beta.eq.sone) then else if (beta.eq.sone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) + y(i,j) y(i,j) = x(i,j) + y(i,j)
enddo enddo
@ -516,12 +521,14 @@ subroutine saxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-sone) then else if (beta.eq.-sone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) - y(i,j) y(i,j) = x(i,j) - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) + beta*y(i,j) y(i,j) = x(i,j) + beta*y(i,j)
enddo enddo
@ -532,12 +539,14 @@ subroutine saxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (beta.eq.szero) then if (beta.eq.szero) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) y(i,j) = -x(i,j)
enddo enddo
enddo enddo
else if (beta.eq.sone) then else if (beta.eq.sone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) + y(i,j) y(i,j) = -x(i,j) + y(i,j)
enddo enddo
@ -545,12 +554,14 @@ subroutine saxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-sone) then else if (beta.eq.-sone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) - y(i,j) y(i,j) = -x(i,j) - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) + beta*y(i,j) y(i,j) = -x(i,j) + beta*y(i,j)
enddo enddo
@ -561,12 +572,14 @@ subroutine saxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (beta.eq.szero) then if (beta.eq.szero) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) y(i,j) = alpha*x(i,j)
enddo enddo
enddo enddo
else if (beta.eq.sone) then else if (beta.eq.sone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) + y(i,j) y(i,j) = alpha*x(i,j) + y(i,j)
enddo enddo
@ -574,12 +587,14 @@ subroutine saxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-sone) then else if (beta.eq.-sone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) - y(i,j) y(i,j) = alpha*x(i,j) - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) + beta*y(i,j) y(i,j) = alpha*x(i,j) + beta*y(i,j)
enddo enddo

15
base/serial/psi_z_serial_impl.f90
Unescape Escape View File

@ -476,6 +476,7 @@ subroutine zaxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (alpha.eq.zzero) then if (alpha.eq.zzero) then
if (beta.eq.zzero) then if (beta.eq.zzero) then
do j=1, n do j=1, n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = zzero y(i,j) = zzero
enddo enddo
@ -487,12 +488,14 @@ subroutine zaxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-zone) then else if (beta.eq.-zone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = - y(i,j) y(i,j) = - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = beta*y(i,j) y(i,j) = beta*y(i,j)
enddo enddo
@ -503,12 +506,14 @@ subroutine zaxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (beta.eq.zzero) then if (beta.eq.zzero) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) y(i,j) = x(i,j)
enddo enddo
enddo enddo
else if (beta.eq.zone) then else if (beta.eq.zone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) + y(i,j) y(i,j) = x(i,j) + y(i,j)
enddo enddo
@ -516,12 +521,14 @@ subroutine zaxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-zone) then else if (beta.eq.-zone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) - y(i,j) y(i,j) = x(i,j) - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = x(i,j) + beta*y(i,j) y(i,j) = x(i,j) + beta*y(i,j)
enddo enddo
@ -532,12 +539,14 @@ subroutine zaxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (beta.eq.zzero) then if (beta.eq.zzero) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) y(i,j) = -x(i,j)
enddo enddo
enddo enddo
else if (beta.eq.zone) then else if (beta.eq.zone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) + y(i,j) y(i,j) = -x(i,j) + y(i,j)
enddo enddo
@ -545,12 +554,14 @@ subroutine zaxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-zone) then else if (beta.eq.-zone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) - y(i,j) y(i,j) = -x(i,j) - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = -x(i,j) + beta*y(i,j) y(i,j) = -x(i,j) + beta*y(i,j)
enddo enddo
@ -561,12 +572,14 @@ subroutine zaxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
if (beta.eq.zzero) then if (beta.eq.zzero) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) y(i,j) = alpha*x(i,j)
enddo enddo
enddo enddo
else if (beta.eq.zone) then else if (beta.eq.zone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) + y(i,j) y(i,j) = alpha*x(i,j) + y(i,j)
enddo enddo
@ -574,12 +587,14 @@ subroutine zaxpby(m, n, alpha, X, lldx, beta, Y, lldy, info)
else if (beta.eq.-zone) then else if (beta.eq.-zone) then
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) - y(i,j) y(i,j) = alpha*x(i,j) - y(i,j)
enddo enddo
enddo enddo
else else
do j=1,n do j=1,n
!$omp parallel do private(i) shared(j)
do i=1,m do i=1,m
y(i,j) = alpha*x(i,j) + beta*y(i,j) y(i,j) = alpha*x(i,j) + beta*y(i,j)
enddo enddo

1
prec/impl/psb_c_diagprec_impl.f90
Unescape Escape View File

@ -255,6 +255,7 @@ subroutine psb_c_diag_precbld(a,desc_a,prec,info,amold,vmold,imold)
end if end if
call psb_realloc(desc_a%get_local_cols(),prec%d,info,pad=cone) call psb_realloc(desc_a%get_local_cols(),prec%d,info,pad=cone)
!$omp parallel do private(i)
do i=1,nrow do i=1,nrow
if (prec%d(i) == dzero) then if (prec%d(i) == dzero) then
prec%d(i) = cone prec%d(i) = cone

1
prec/impl/psb_d_diagprec_impl.f90
Unescape Escape View File

@ -255,6 +255,7 @@ subroutine psb_d_diag_precbld(a,desc_a,prec,info,amold,vmold,imold)
end if end if
call psb_realloc(desc_a%get_local_cols(),prec%d,info,pad=done) call psb_realloc(desc_a%get_local_cols(),prec%d,info,pad=done)
!$omp parallel do private(i)
do i=1,nrow do i=1,nrow
if (prec%d(i) == dzero) then if (prec%d(i) == dzero) then
prec%d(i) = done prec%d(i) = done

1
prec/impl/psb_s_diagprec_impl.f90
Unescape Escape View File

@ -255,6 +255,7 @@ subroutine psb_s_diag_precbld(a,desc_a,prec,info,amold,vmold,imold)
end if end if
call psb_realloc(desc_a%get_local_cols(),prec%d,info,pad=sone) call psb_realloc(desc_a%get_local_cols(),prec%d,info,pad=sone)
!$omp parallel do private(i)
do i=1,nrow do i=1,nrow
if (prec%d(i) == dzero) then if (prec%d(i) == dzero) then
prec%d(i) = sone prec%d(i) = sone

1
prec/impl/psb_z_diagprec_impl.f90
Unescape Escape View File

@ -255,6 +255,7 @@ subroutine psb_z_diag_precbld(a,desc_a,prec,info,amold,vmold,imold)
end if end if
call psb_realloc(desc_a%get_local_cols(),prec%d,info,pad=zone) call psb_realloc(desc_a%get_local_cols(),prec%d,info,pad=zone)
!$omp parallel do private(i)
do i=1,nrow do i=1,nrow
if (prec%d(i) == dzero) then if (prec%d(i) == dzero) then
prec%d(i) = zone prec%d(i) = zone

Write Preview
Loading…
Cancel
Save