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

Final fix for COO on OMP

Browse Source
This commit is contained in:
sfilippone
2023-08-22 10:22:46 +02:00
parent 26bf4c5d69
commit f001ebbad3
4 changed files with 32 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
base/serial/impl/psb_c_coo_impl.F90
+8 -6
View File
@@ -2864,7 +2864,9 @@ subroutine psb_c_coo_csput_a(nz,ia,ja,val,a,imin,imax,jmin,jmax,info)
if (nz == 0) return
if (a%is_bld()) then
! Structure here is peculiar, because this function can be called
! either within a parallel region, or outside.
! Hence the call to set_nzeros done here.
!$omp critical
nza = a%get_nzeros()
isza = a%get_size()
@@ -2875,14 +2877,15 @@ subroutine psb_c_coo_csput_a(nz,ia,ja,val,a,imin,imax,jmin,jmax,info)
isza = a%get_size()
if (isza < (nza+nz)) then
info = psb_err_alloc_dealloc_; call psb_errpush(info,name)
else
nzaold = nza
nza = nza + nz
call a%set_nzeros(nza)
end if
!$omp end critical
if (info /= 0) goto 9999
call psb_inner_ins(nz,ia,ja,val,nza,a%ia,a%ja,a%val,isza,&
call psb_inner_ins(nz,ia,ja,val,nzaold,a%ia,a%ja,a%val,isza,&
& imin,imax,jmin,jmax,info)
!write(0,*) 'From CSPUT :',nza,nzaold
call a%set_nzeros(nza)
call a%set_sorted(.false.)
else if (a%is_upd()) then
@@ -2956,7 +2959,6 @@ contains
end if
end do
!$OMP END PARALLEL DO
nza = nza + nz
#else
do i=1, nz
base/serial/impl/psb_d_coo_impl.F90
+8 -6
View File
@@ -2864,7 +2864,9 @@ subroutine psb_d_coo_csput_a(nz,ia,ja,val,a,imin,imax,jmin,jmax,info)
if (nz == 0) return
if (a%is_bld()) then
! Structure here is peculiar, because this function can be called
! either within a parallel region, or outside.
! Hence the call to set_nzeros done here.
!$omp critical
nza = a%get_nzeros()
isza = a%get_size()
@@ -2875,14 +2877,15 @@ subroutine psb_d_coo_csput_a(nz,ia,ja,val,a,imin,imax,jmin,jmax,info)
isza = a%get_size()
if (isza < (nza+nz)) then
info = psb_err_alloc_dealloc_; call psb_errpush(info,name)
else
nzaold = nza
nza = nza + nz
call a%set_nzeros(nza)
end if
!$omp end critical
if (info /= 0) goto 9999
call psb_inner_ins(nz,ia,ja,val,nza,a%ia,a%ja,a%val,isza,&
call psb_inner_ins(nz,ia,ja,val,nzaold,a%ia,a%ja,a%val,isza,&
& imin,imax,jmin,jmax,info)
!write(0,*) 'From CSPUT :',nza,nzaold
call a%set_nzeros(nza)
call a%set_sorted(.false.)
else if (a%is_upd()) then
@@ -2956,7 +2959,6 @@ contains
end if
end do
!$OMP END PARALLEL DO
nza = nza + nz
#else
do i=1, nz
base/serial/impl/psb_s_coo_impl.F90
+8 -6
View File
@@ -2864,7 +2864,9 @@ subroutine psb_s_coo_csput_a(nz,ia,ja,val,a,imin,imax,jmin,jmax,info)
if (nz == 0) return
if (a%is_bld()) then
! Structure here is peculiar, because this function can be called
! either within a parallel region, or outside.
! Hence the call to set_nzeros done here.
!$omp critical
nza = a%get_nzeros()
isza = a%get_size()
@@ -2875,14 +2877,15 @@ subroutine psb_s_coo_csput_a(nz,ia,ja,val,a,imin,imax,jmin,jmax,info)
isza = a%get_size()
if (isza < (nza+nz)) then
info = psb_err_alloc_dealloc_; call psb_errpush(info,name)
else
nzaold = nza
nza = nza + nz
call a%set_nzeros(nza)
end if
!$omp end critical
if (info /= 0) goto 9999
call psb_inner_ins(nz,ia,ja,val,nza,a%ia,a%ja,a%val,isza,&
call psb_inner_ins(nz,ia,ja,val,nzaold,a%ia,a%ja,a%val,isza,&
& imin,imax,jmin,jmax,info)
!write(0,*) 'From CSPUT :',nza,nzaold
call a%set_nzeros(nza)
call a%set_sorted(.false.)
else if (a%is_upd()) then
@@ -2956,7 +2959,6 @@ contains
end if
end do
!$OMP END PARALLEL DO
nza = nza + nz
#else
do i=1, nz
base/serial/impl/psb_z_coo_impl.F90
+8 -6
View File
@@ -2864,7 +2864,9 @@ subroutine psb_z_coo_csput_a(nz,ia,ja,val,a,imin,imax,jmin,jmax,info)
if (nz == 0) return
if (a%is_bld()) then
! Structure here is peculiar, because this function can be called
! either within a parallel region, or outside.
! Hence the call to set_nzeros done here.
!$omp critical
nza = a%get_nzeros()
isza = a%get_size()
@@ -2875,14 +2877,15 @@ subroutine psb_z_coo_csput_a(nz,ia,ja,val,a,imin,imax,jmin,jmax,info)
isza = a%get_size()
if (isza < (nza+nz)) then
info = psb_err_alloc_dealloc_; call psb_errpush(info,name)
else
nzaold = nza
nza = nza + nz
call a%set_nzeros(nza)
end if
!$omp end critical
if (info /= 0) goto 9999
call psb_inner_ins(nz,ia,ja,val,nza,a%ia,a%ja,a%val,isza,&
call psb_inner_ins(nz,ia,ja,val,nzaold,a%ia,a%ja,a%val,isza,&
& imin,imax,jmin,jmax,info)
!write(0,*) 'From CSPUT :',nza,nzaold
call a%set_nzeros(nza)
call a%set_sorted(.false.)
else if (a%is_upd()) then
@@ -2956,7 +2959,6 @@ contains
end if
end do
!$OMP END PARALLEL DO
nza = nza + nz
#else
do i=1, nz