Homec4science
Diffusion SPClibs (master) Browse

spclibs/examples/tpsum_mat.f90
93a1e9712d04master

Last Change
Raw File

tpsum_mat.f90
View Options

!>
!> @file tpsum_mat.f90
!>
!> @brief
!>
!> @copyright
!> Copyright (©) 2021 EPFL (Ecole Polytechnique Fédérale de Lausanne)
!> SPC (Swiss Plasma Center)
!>
!> SPClibs is free software: you can redistribute it and/or modify it under
!> the terms of the GNU Lesser General Public License as published by the Free
!> Software Foundation, either version 3 of the License, or (at your option)
!> any later version.
!>
!> SPClibs is distributed in the hope that it will be useful, but WITHOUT ANY
!> WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
!> FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
!>
!> You should have received a copy of the GNU Lesser General Public License
!> along with this program. If not, see <https://www.gnu.org/licenses/>.
!>
!> @authors
!> (in alphabetical order)
!> @author Trach-Minh Tran <trach-minh.tran@epfl.ch>
!>
PROGRAM main
!!$ USE pardiso_bsplines
!!$ USE wsmp_bsplines
USE mumps_bsplines
IMPLICIT NONE
INCLUDE 'mpif.h'
!
INTEGER :: me, npes, ierr
INTEGER :: i, j, rank
DOUBLE PRECISION :: val
DOUBLE PRECISION, ALLOCATABLE :: arow(:)
!!$ TYPE(pardiso_mat) :: mat
!!$ TYPE(wsmp_mat) :: mat
TYPE(mumps_mat) :: mat
!
CALL mpi_init(ierr)
CALL mpi_comm_size(MPI_COMM_WORLD, npes, ierr)
CALL mpi_comm_rank(MPI_COMM_WORLD, me, ierr)
!
rank = npes
CALL init(npes, 1, mat)
DO i=1,rank ! Fill row me+1
val = me+1
j = me+1
CALL updtmat(mat, i, me+1, val)
END DO
!
!!$ CALL disp_mat('Original matrix')
CALL psum_mat(mat, MPI_COMM_WORLD)
CALL disp_mat('Global sum of matrix')
!
CALL mpi_finalize(ierr)
CONTAINS
SUBROUTINE disp_mat(str)
CHARACTER(len=*), INTENT(in) :: str
INTEGER :: p
DO p=0,npes-1
IF(me.EQ.p) THEN
CALL to_mat(mat, nlkeep=.TRUE.)
WRITE(*,'(a,i3.3,a,2i6)') str//' on PE', me, ': rank, nnz', mat%rank, mat%nnz
ALLOCATE(arow(mat%rank))
DO i=1,mat%rank
CALL getrow(mat, i, arow)
WRITE(*,'(10f8.2)') arow
END DO
DEALLOCATE(arow)
CALL FLUSH(6)
END IF
CALL mpi_barrier(MPI_COMM_WORLD, ierr)
END DO
END SUBROUTINE disp_mat
END PROGRAM main
c4science · Help