Reduce_timestep() [2/7]

template<typename T >

void cntr::Reduce_timestep	(	int	tstp,
		int	root,
		herm_matrix_timestep_view< T > &	Gred,
		herm_matrix_timestep_view< T > &	G
	)

MPI reduce for the herm_matrix_timestep_view to a herm_matrix_timestep_view

Purpose

MPI reduce for the herm_matrix_timestep_view to the root Works for scalar or square-matrix contour objects.

Parameters

tstp	time step
root	Index of root
Gred	The reduced herm_matrix_timestep_view on rank root.
G	The herm_matrix_timestep_view on the individual ranks.

Definition at line 82 of file cntr_mpitools_impl.hpp.

References cntr::herm_matrix_timestep_view< T >::ntau(), cntr::herm_matrix_timestep_view< T >::size1(), cntr::herm_matrix_timestep_view< T >::size2(), and cntr::herm_matrix_timestep_view< T >::tstp().

                                    {
    assert(tstp == G.tstp());
    int taskid;
    MPI_Comm_rank(MPI_COMM_WORLD, &taskid);
    if (taskid == root) {
        assert(tstp == Gred.tstp());
        assert(G.ntau() == Gred.ntau());
        assert(G.size1() == Gred.size1());
        assert(G.size2() == Gred.size2());
    }

    int len_rt = 2 * (tstp + 1) * G.size1() * G.size2();
    int len_it = 2 * (G.ntau() + 1) * G.size1() * G.size2();

    if (sizeof(T) == sizeof(double)) {
        if(tstp == -1){
            MPI_Reduce((double *)G.mat_, (double *)Gred.mat_, len_it, MPI_DOUBLE_PRECISION, MPI_SUM, root,
                MPI_COMM_WORLD);
        } else{
            MPI_Reduce((double *)G.les_, (double *)Gred.les_, len_rt, MPI_DOUBLE_PRECISION, MPI_SUM, root,
                MPI_COMM_WORLD);
            MPI_Reduce((double *)G.ret_, (double *)Gred.ret_, len_rt, MPI_DOUBLE_PRECISION, MPI_SUM, root,
                MPI_COMM_WORLD);
            MPI_Reduce((double *)G.tv_, (double *)Gred.tv_, len_it, MPI_DOUBLE_PRECISION, MPI_SUM, root,
                MPI_COMM_WORLD);
        }
        
   } else {
      if (taskid == root) std::cerr << "herm_matrix_timestep_view<T>::MPI_Reduce only for double " << std::endl;
      MPI_Finalize();
      exit(0);
   }

}

Here is the call graph for this function: