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

Setter/getter for SPSPMM algorithm in base_mat_mod

Browse Source
This commit is contained in:
Salvatore Filippone
2023-08-29 05:07:55 -04:00
parent 2322a9ce61
commit 7b45994b70
2 changed files with 30 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
base/modules/serial/psb_base_mat_mod.F90
+28
View File
@@ -73,6 +73,16 @@ module psb_base_mat_mod
use psb_const_mod
use psi_serial_mod
! Algs for SPSPMM
integer(psb_ipk_), parameter :: spspmm_serial = 0
integer(psb_ipk_), parameter :: spspmm_omp_gustavson = 1
integer(psb_ipk_), parameter :: spspmm_omp_gustavson_1d = 2
integer(psb_ipk_), parameter :: spspmm_serial_rb_tree = 3
integer(psb_ipk_), parameter :: spspmm_omp_rb_tree = 4
integer(psb_ipk_), parameter :: spspmm_omp_two_pass = 5
integer(psb_ipk_), save :: spspmm_impl = spspmm_serial
!
!> \namespace psb_base_mod \class psb_base_sparse_mat
!! The basic data about your matrix.
@@ -1904,4 +1914,22 @@ contains
end subroutine psb_base_from_lbase
function psb_get_spspmm_impl() result(impl_id)
integer(psb_ipk_) :: impl_id
impl_id = spspmm_impl
end function psb_get_spspmm_impl
subroutine psb_set_spspmm_impl(impl_id)
integer(psb_ipk_), intent(in) :: impl_id
select case(impl_id)
case (spspmm_serial,spspmm_omp_gustavson,spspmm_omp_gustavson_1d,&
& spspmm_serial_rb_tree,spspmm_omp_rb_tree, spspmm_omp_two_pass)
spspmm_impl = impl_id
case default
write (*,*) "Invalid implementation id",impl_id
end select
end subroutine psb_set_spspmm_impl
end module psb_base_mat_mod
base/modules/serial/psb_d_csr_mat_mod.f90
+2 -22
View File
@@ -45,15 +45,6 @@
module psb_d_csr_mat_mod
use psb_d_base_mat_mod
integer(psb_ipk_), parameter :: spspmm_serial = 0
integer(psb_ipk_), parameter :: spspmm_omp_gustavson = 1
integer(psb_ipk_), parameter :: spspmm_omp_gustavson_1d = 2
integer(psb_ipk_), parameter :: spspmm_serial_rb_tree = 3
integer(psb_ipk_), parameter :: spspmm_omp_rb_tree = 4
integer(psb_ipk_), parameter :: spspmm_omp_two_pass = 5
integer(psb_ipk_) :: spspmm_impl = spspmm_serial
!> \namespace psb_base_mod \class psb_d_csr_sparse_mat
!! \extends psb_d_base_mat_mod::psb_d_base_sparse_mat
!!
@@ -586,8 +577,8 @@ module psb_d_csr_mat_mod
integer(psb_ipk_), intent(out) :: info
end subroutine psb_d_csr_scals
end interface
!> \namespace psb_base_mod \class psb_ld_csr_sparse_mat
!> \namespace psb_base_mod \class psb_ld_csr_sparse_mat
!! \extends psb_ld_base_mat_mod::psb_ld_base_sparse_mat
!!
!! psb_ld_csr_sparse_mat type and the related methods.
@@ -1301,15 +1292,4 @@ contains
end subroutine ld_csr_free
subroutine set_d_csr_spspmm_impl(impl_id)
integer(psb_ipk_), intent(in) :: impl_id
if (impl_id < 0 .or. impl_id > 5) then
write (*,*) "Invalid implementation id, impl id set to serial"
spspmm_impl = spspmm_serial
else
spspmm_impl = impl_id
end if
end subroutine
end module psb_d_csr_mat_mod