fdurastante/psblas3
1
0
Fork 0
mirror of https://github.com/sfilippone/psblas3.git synced 2026-10-20 12:52:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'development' of github.com:sfilippone/psblas3 into development

Browse Source
This commit is contained in:
Salvatore Filippone
2021-01-26 17:05:29 +01:00
parent 7d200ccc33 73605bdda0
commit 50da50f557
4 changed files with 152 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files
base/modules/serial/psb_c_vect_mod.F90
+38 -7
View File
@@ -1028,9 +1028,23 @@ contains
real(psb_spk_) :: res
integer(psb_ipk_) :: info
if (allocated(x%v).and.allocated(w%v)) then
call w%v%mlt(x%v,info)
res = w%v%nrm2(n)
! Temp vectors
type(psb_c_vect_type) :: wtemp
if( allocated(w%v) ) then
! FIXME for GPU
allocate(wtemp%v, source=w%v, stat = info)
else
info = -1
end if
if (info /= 0 ) then
res = -sone
return
end if
if (allocated(x%v)) then
call wtemp%v%mlt(x%v,info)
res = wtemp%v%nrm2(n)
else
res = szero
end if
@@ -1046,10 +1060,27 @@ contains
real(psb_spk_) :: res
integer(psb_ipk_) :: info
if (allocated(x%v).and.allocated(w%v).and.allocated(id%v)) then
where( abs(id%v%v) <= szero) x%v%v = szero
call w%v%mlt(x%v,info)
res = w%v%nrm2(n)
! Temp vectors
type(psb_c_vect_type) :: wtemp
if( allocated(w%v) ) then
! FIXME for GPU
allocate(wtemp%v, source=w%v, stat = info)
else
info = -1
end if
if (info /= 0 ) then
res = -sone
return
end if
if (allocated(x%v).and.allocated(id%v)) then
call wtemp%sync() ! FIXME for GPU
where( abs(id%v%v) <= szero) wtemp%v%v = szero
call wtemp%set_host() ! FIXME for GPU
call wtemp%v%mlt(x%v,info)
res = wtemp%v%nrm2(n)
else
res = szero
end if
base/modules/serial/psb_d_vect_mod.F90
+38 -7
View File
@@ -1035,9 +1035,23 @@ contains
real(psb_dpk_) :: res
integer(psb_ipk_) :: info
if (allocated(x%v).and.allocated(w%v)) then
call w%v%mlt(x%v,info)
res = w%v%nrm2(n)
! Temp vectors
type(psb_d_vect_type) :: wtemp
if( allocated(w%v) ) then
! FIXME for GPU
allocate(wtemp%v, source=w%v, stat = info)
else
info = -1
end if
if (info /= 0 ) then
res = -done
return
end if
if (allocated(x%v)) then
call wtemp%v%mlt(x%v,info)
res = wtemp%v%nrm2(n)
else
res = dzero
end if
@@ -1053,10 +1067,27 @@ contains
real(psb_dpk_) :: res
integer(psb_ipk_) :: info
if (allocated(x%v).and.allocated(w%v).and.allocated(id%v)) then
where( abs(id%v%v) <= dzero) x%v%v = dzero
call w%v%mlt(x%v,info)
res = w%v%nrm2(n)
! Temp vectors
type(psb_d_vect_type) :: wtemp
if( allocated(w%v) ) then
! FIXME for GPU
allocate(wtemp%v, source=w%v, stat = info)
else
info = -1
end if
if (info /= 0 ) then
res = -done
return
end if
if (allocated(x%v).and.allocated(id%v)) then
call wtemp%sync() ! FIXME for GPU
where( abs(id%v%v) <= dzero) wtemp%v%v = dzero
call wtemp%set_host() ! FIXME for GPU
call wtemp%v%mlt(x%v,info)
res = wtemp%v%nrm2(n)
else
res = dzero
end if
base/modules/serial/psb_s_vect_mod.F90
+38 -7
View File
@@ -1035,9 +1035,23 @@ contains
real(psb_spk_) :: res
integer(psb_ipk_) :: info
if (allocated(x%v).and.allocated(w%v)) then
call w%v%mlt(x%v,info)
res = w%v%nrm2(n)
! Temp vectors
type(psb_s_vect_type) :: wtemp
if( allocated(w%v) ) then
! FIXME for GPU
allocate(wtemp%v, source=w%v, stat = info)
else
info = -1
end if
if (info /= 0 ) then
res = -sone
return
end if
if (allocated(x%v)) then
call wtemp%v%mlt(x%v,info)
res = wtemp%v%nrm2(n)
else
res = szero
end if
@@ -1053,10 +1067,27 @@ contains
real(psb_spk_) :: res
integer(psb_ipk_) :: info
if (allocated(x%v).and.allocated(w%v).and.allocated(id%v)) then
where( abs(id%v%v) <= szero) x%v%v = szero
call w%v%mlt(x%v,info)
res = w%v%nrm2(n)
! Temp vectors
type(psb_s_vect_type) :: wtemp
if( allocated(w%v) ) then
! FIXME for GPU
allocate(wtemp%v, source=w%v, stat = info)
else
info = -1
end if
if (info /= 0 ) then
res = -sone
return
end if
if (allocated(x%v).and.allocated(id%v)) then
call wtemp%sync() ! FIXME for GPU
where( abs(id%v%v) <= szero) wtemp%v%v = szero
call wtemp%set_host() ! FIXME for GPU
call wtemp%v%mlt(x%v,info)
res = wtemp%v%nrm2(n)
else
res = szero
end if
base/modules/serial/psb_z_vect_mod.F90
+38 -7
View File
@@ -1028,9 +1028,23 @@ contains
real(psb_dpk_) :: res
integer(psb_ipk_) :: info
if (allocated(x%v).and.allocated(w%v)) then
call w%v%mlt(x%v,info)
res = w%v%nrm2(n)
! Temp vectors
type(psb_z_vect_type) :: wtemp
if( allocated(w%v) ) then
! FIXME for GPU
allocate(wtemp%v, source=w%v, stat = info)
else
info = -1
end if
if (info /= 0 ) then
res = -done
return
end if
if (allocated(x%v)) then
call wtemp%v%mlt(x%v,info)
res = wtemp%v%nrm2(n)
else
res = dzero
end if
@@ -1046,10 +1060,27 @@ contains
real(psb_dpk_) :: res
integer(psb_ipk_) :: info
if (allocated(x%v).and.allocated(w%v).and.allocated(id%v)) then
where( abs(id%v%v) <= dzero) x%v%v = dzero
call w%v%mlt(x%v,info)
res = w%v%nrm2(n)
! Temp vectors
type(psb_z_vect_type) :: wtemp
if( allocated(w%v) ) then
! FIXME for GPU
allocate(wtemp%v, source=w%v, stat = info)
else
info = -1
end if
if (info /= 0 ) then
res = -done
return
end if
if (allocated(x%v).and.allocated(id%v)) then
call wtemp%sync() ! FIXME for GPU
where( abs(id%v%v) <= dzero) wtemp%v%v = dzero
call wtemp%set_host() ! FIXME for GPU
call wtemp%v%mlt(x%v,info)
res = wtemp%v%nrm2(n)
else
res = dzero
end if