spmf_rebalance.F90

Go to the documentation of this file.

!>
!> @file spmf_rebalance.F90
!>
!> @brief Fortran 90 internal testing to check gather/Scatter fortran
!> interface.
!>
!> It basically gather a non balanced distributed spm, and
!> re-distribute it in a balanced way.
!>
!> @copyright 2017-2024 Bordeaux INP, CNRS (LaBRI UMR 5800), Inria,
!>                      Univ. Bordeaux. All rights reserved.
!>
!> @version 1.2.4
!> @author Mathieu Faverge
!> @author Alycia Lisito
!> @date 2024-05-29
!>
!> @ingroup examples_fortran
!>
#ifndef DOXYGEN_SHOULD_SKIP_THIS
program spmf_rebalance
  use iso_c_binding
  use spmf
#if defined(SPM_WITH_MPI)
  use mpi_f08
#endif
  implicit none
 
  type(spmatrix_t), target :: spm
  type(spmatrix_t), target :: spml
  type(spmatrix_t), target :: spmg
  integer(c_int)           :: info
  integer(spm_int_t)       :: n
  integer                  :: rank = 0
  integer                  :: size = 1
 
#if defined(SPM_WITH_MPI)
  ! SPM is compiled with MPI, thus MPI must be initialized
  ! before any spm call
  call mpi_init( info )
  call mpi_comm_rank( mpi_comm_world, rank )
  call mpi_comm_size( mpi_comm_world, size )
#endif
 
  !
  ! Initialize the problem
  !   1- The matrix
  call spmreaddriverdist( spmdriverlaplacian, "d:10:10:10:4.", spm, mpi_comm_world, info )
  call spmprintinfo( spm )
 
  n = spm%gN / size
 
  if ( rank .eq. 0 ) then
     call spmgather( spm, 0, spmg )
     call spmprintinfo( spmg )
  else
     call spmgather( spm, 0 )
  end if
 
  call spmexit( spm )
 
  if ( rank .eq. 0 ) then
     n = n + mod( spm%gN, size )
     call spmscatter( spml, 0, spmg )
     call spmexit( spmg )
  else
     call spmscatter( spml, 0 )
  end if
 
  call spmprintinfo( spml )
  call spmexit( spml )
 
#if defined(SPM_WITH_MPI)
  call mpi_finalize( info )
#endif
 
end program spmf_rebalance
 
#endif