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

Use less memory in spspmm

Browse Source
pull/28/head
sfilippone 11 months ago
parent 67b2937e6e
commit 86a6ab413a
4 changed files with 24 additions and 24 deletions
Show Stats Download Patch File Download Diff File

12
base/serial/impl/psb_c_csr_impl.F90
Unescape Escape View File

@ -3693,7 +3693,7 @@ subroutine psb_ccsrspspmm(a,b,c,info)
! Estimate number of nonzeros on output.
nza = a%get_nzeros()
nzb = b%get_nzeros()
nzc = 2*(nza+nzb)
nzc = max(nint(0.5*(nza+nzb)),ma,mb,na,nb)
call c%allocate(ma,nb,nzc)
call csr_spspmm(a,b,c,info)
@ -3773,8 +3773,8 @@ contains
if (nrc > 0 ) then
if ((nzc+nrc)>nze) then
nze = max(ma*((nzc+j-1)/j),nzc+2*nrc)
call psb_realloc(nze,c%val,info)
if (info == 0) call psb_realloc(nze,c%ja,info)
call psb_ensure_size(nze,c%val,info)
if (info == 0) call psb_ensure_size(nze,c%ja,info)
if (info /= 0) return
end if
@ -4238,7 +4238,7 @@ subroutine psb_ccsrspspmm(a,b,c,info)
! Estimate number of nonzeros on output.
nza = a%get_nzeros()
nzb = b%get_nzeros()
nzc = 2*(nza+nzb)
nzc = max(nint(0.5*(nza+nzb)),ma,mb,na,nb)
call c%allocate(ma,nb,nzc)
call csr_spspmm(a,b,c,info)
@ -4303,8 +4303,8 @@ contains
if (nrc > 0 ) then
if ((nzc+nrc)>nze) then
nze = max(ma*((nzc+j-1)/j),nzc+2*nrc)
call psb_realloc(nze,c%val,info)
if (info == 0) call psb_realloc(nze,c%ja,info)
call psb_ensure_size(nze,c%val,info)
if (info == 0) call psb_ensure_size(nze,c%ja,info)
if (info /= 0) return
end if

12
base/serial/impl/psb_d_csr_impl.F90
Unescape Escape View File

@ -3693,7 +3693,7 @@ subroutine psb_dcsrspspmm(a,b,c,info)
! Estimate number of nonzeros on output.
nza = a%get_nzeros()
nzb = b%get_nzeros()
nzc = 2*(nza+nzb)
nzc = max(nint(0.5*(nza+nzb)),ma,mb,na,nb)
call c%allocate(ma,nb,nzc)
call csr_spspmm(a,b,c,info)
@ -3773,8 +3773,8 @@ contains
if (nrc > 0 ) then
if ((nzc+nrc)>nze) then
nze = max(ma*((nzc+j-1)/j),nzc+2*nrc)
call psb_realloc(nze,c%val,info)
if (info == 0) call psb_realloc(nze,c%ja,info)
call psb_ensure_size(nze,c%val,info)
if (info == 0) call psb_ensure_size(nze,c%ja,info)
if (info /= 0) return
end if
@ -4238,7 +4238,7 @@ subroutine psb_dcsrspspmm(a,b,c,info)
! Estimate number of nonzeros on output.
nza = a%get_nzeros()
nzb = b%get_nzeros()
nzc = 2*(nza+nzb)
nzc = max(nint(0.5*(nza+nzb)),ma,mb,na,nb)
call c%allocate(ma,nb,nzc)
call csr_spspmm(a,b,c,info)
@ -4303,8 +4303,8 @@ contains
if (nrc > 0 ) then
if ((nzc+nrc)>nze) then
nze = max(ma*((nzc+j-1)/j),nzc+2*nrc)
call psb_realloc(nze,c%val,info)
if (info == 0) call psb_realloc(nze,c%ja,info)
call psb_ensure_size(nze,c%val,info)
if (info == 0) call psb_ensure_size(nze,c%ja,info)
if (info /= 0) return
end if

12
base/serial/impl/psb_s_csr_impl.F90
Unescape Escape View File

@ -3693,7 +3693,7 @@ subroutine psb_scsrspspmm(a,b,c,info)
! Estimate number of nonzeros on output.
nza = a%get_nzeros()
nzb = b%get_nzeros()
nzc = 2*(nza+nzb)
nzc = max(nint(0.5*(nza+nzb)),ma,mb,na,nb)
call c%allocate(ma,nb,nzc)
call csr_spspmm(a,b,c,info)
@ -3773,8 +3773,8 @@ contains
if (nrc > 0 ) then
if ((nzc+nrc)>nze) then
nze = max(ma*((nzc+j-1)/j),nzc+2*nrc)
call psb_realloc(nze,c%val,info)
if (info == 0) call psb_realloc(nze,c%ja,info)
call psb_ensure_size(nze,c%val,info)
if (info == 0) call psb_ensure_size(nze,c%ja,info)
if (info /= 0) return
end if
@ -4238,7 +4238,7 @@ subroutine psb_scsrspspmm(a,b,c,info)
! Estimate number of nonzeros on output.
nza = a%get_nzeros()
nzb = b%get_nzeros()
nzc = 2*(nza+nzb)
nzc = max(nint(0.5*(nza+nzb)),ma,mb,na,nb)
call c%allocate(ma,nb,nzc)
call csr_spspmm(a,b,c,info)
@ -4303,8 +4303,8 @@ contains
if (nrc > 0 ) then
if ((nzc+nrc)>nze) then
nze = max(ma*((nzc+j-1)/j),nzc+2*nrc)
call psb_realloc(nze,c%val,info)
if (info == 0) call psb_realloc(nze,c%ja,info)
call psb_ensure_size(nze,c%val,info)
if (info == 0) call psb_ensure_size(nze,c%ja,info)
if (info /= 0) return
end if

12
base/serial/impl/psb_z_csr_impl.F90
Unescape Escape View File

@ -3693,7 +3693,7 @@ subroutine psb_zcsrspspmm(a,b,c,info)
! Estimate number of nonzeros on output.
nza = a%get_nzeros()
nzb = b%get_nzeros()
nzc = 2*(nza+nzb)
nzc = max(nint(0.5*(nza+nzb)),ma,mb,na,nb)
call c%allocate(ma,nb,nzc)
call csr_spspmm(a,b,c,info)
@ -3773,8 +3773,8 @@ contains
if (nrc > 0 ) then
if ((nzc+nrc)>nze) then
nze = max(ma*((nzc+j-1)/j),nzc+2*nrc)
call psb_realloc(nze,c%val,info)
if (info == 0) call psb_realloc(nze,c%ja,info)
call psb_ensure_size(nze,c%val,info)
if (info == 0) call psb_ensure_size(nze,c%ja,info)
if (info /= 0) return
end if
@ -4238,7 +4238,7 @@ subroutine psb_zcsrspspmm(a,b,c,info)
! Estimate number of nonzeros on output.
nza = a%get_nzeros()
nzb = b%get_nzeros()
nzc = 2*(nza+nzb)
nzc = max(nint(0.5*(nza+nzb)),ma,mb,na,nb)
call c%allocate(ma,nb,nzc)
call csr_spspmm(a,b,c,info)
@ -4303,8 +4303,8 @@ contains
if (nrc > 0 ) then
if ((nzc+nrc)>nze) then
nze = max(ma*((nzc+j-1)/j),nzc+2*nrc)
call psb_realloc(nze,c%val,info)
if (info == 0) call psb_realloc(nze,c%ja,info)
call psb_ensure_size(nze,c%val,info)
if (info == 0) call psb_ensure_size(nze,c%ja,info)
if (info /= 0) return
end if

Write Preview
Loading…
Cancel
Save