Setter/getter for SPSPMM algorithm in base_mat_mod

omp-walther
Salvatore Filippone 1 year ago
parent 2322a9ce61
commit 7b45994b70

@ -73,6 +73,16 @@ module psb_base_mat_mod
use psb_const_mod use psb_const_mod
use psi_serial_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 !> \namespace psb_base_mod \class psb_base_sparse_mat
!! The basic data about your matrix. !! The basic data about your matrix.
@ -1904,4 +1914,22 @@ contains
end subroutine psb_base_from_lbase 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 end module psb_base_mat_mod

@ -45,15 +45,6 @@
module psb_d_csr_mat_mod module psb_d_csr_mat_mod
use psb_d_base_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 !> \namespace psb_base_mod \class psb_d_csr_sparse_mat
!! \extends psb_d_base_mat_mod::psb_d_base_sparse_mat !! \extends psb_d_base_mat_mod::psb_d_base_sparse_mat
!! !!
@ -1301,15 +1292,4 @@ contains
end subroutine ld_csr_free 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 end module psb_d_csr_mat_mod

Loading…
Cancel
Save