MpiML2DBase< T > Class Template Reference

#include <mpi_ml_align2d.h>

Collaboration diagram for MpiML2DBase< T >:

Public Member Functions
void	readMpi (int argc, char **argv)
	MpiML2DBase (T *prm)
	MpiML2DBase (T *prm, MpiNode *node)
	MpiML2DBase (const MpiML2DBase &)=delete
	MpiML2DBase (const MpiML2DBase &&)=delete
	~MpiML2DBase ()
MpiML2DBase &	operator= (const MpiML2DBase &)=delete
MpiML2DBase &	operator= (const MpiML2DBase &&)=delete
void	sendDocfile (const MultidimArray< double > &data)

Protected Attributes
MpiNode *	node
bool	created_node
T *	program

Detailed Description

template<typename T>
class MpiML2DBase< T >

Class to organize some useful MPI-functions for ML programs It will also serve as base for those programs

Definition at line 41 of file mpi_ml_align2d.h.

Constructor & Destructor Documentation

MpiML2DBase() [1/4]

template<typename T>

MpiML2DBase< T >::MpiML2DBase

(

T *

prm

)

Default constructor

Definition at line 34 of file mpi_ml_align2d.cpp.

{
    node = nullptr;
    program = prm;
}

MpiML2DBase() [2/4]

template<typename T>

MpiML2DBase< T >::MpiML2DBase	(	T *	prm,
		MpiNode *	node
	)

Constructor passing the MpiNode

Definition at line 41 of file mpi_ml_align2d.cpp.

{
    node = mpinode;
    created_node = false;
    program = prm;
}

MpiML2DBase() [3/4]

template<typename T>

MpiML2DBase< T >::MpiML2DBase ( const MpiML2DBase< T > &  )

delete

MpiML2DBase() [4/4]

template<typename T>

MpiML2DBase< T >::MpiML2DBase ( const MpiML2DBase< T > &&  )

delete

~MpiML2DBase()

template<typename T>

MpiML2DBase< T >::~MpiML2DBase ( )

inline

Destructor

Definition at line 80 of file mpi_ml_align2d.h.

                   {
        if (created_node)
            delete node;
    }

Member Function Documentation

operator=() [1/2]

template<typename T>

MpiML2DBase& MpiML2DBase< T >::operator= ( const MpiML2DBase< T > &  )

delete

operator=() [2/2]

template<typename T>

MpiML2DBase& MpiML2DBase< T >::operator= ( const MpiML2DBase< T > &&  )

delete

readMpi()

template<typename T>

void MpiML2DBase< T >::readMpi ( int  argc, char **  argv  )

inline

Read arguments sequentially to avoid concurrency problems

Definition at line 51 of file mpi_ml_align2d.h.

                                        {
        if (node == nullptr)
        {
            node = new MpiNode(argc, argv);
            created_node = true;
        }
        //The following makes the asumption that 'this' also
        //inherits from an XmippProgram
        if (!node->isMaster())
            program->verbose = 0;
        // Read subsequently to avoid problems in restart procedure
        for (size_t proc = 0; proc < node->size; ++proc)
        {
            if (proc == node->rank)
                program->read(argc, (const char **)argv);
            node->barrierWait();
        }
        //Send "master" seed to slaves for same randomization
        MPI_Bcast(&program->seed, 1, MPI_INT, 0, MPI_COMM_WORLD);
    }

sendDocfile()

template<typename T >

void MpiML2DBase< T >::sendDocfile

(

const MultidimArray< double > &

data

)

This function is only valid for 2D ML programs

Definition at line 49 of file mpi_ml_align2d.cpp.

{

    // Write intermediate files
    if (!node->isMaster())
    {
        // All slaves send docfile data to the master
        int s_size = MULTIDIM_SIZE(docfiledata);
        MPI_Send(&s_size, 1, MPI_INT, 0, TAG_DOCFILESIZE,
                 MPI_COMM_WORLD);
        MPI_Send(MULTIDIM_ARRAY(docfiledata), s_size, MPI_DOUBLE,
                 0, TAG_DOCFILE, MPI_COMM_WORLD);
    }
    else
    {
        // Master fills docfile
        // Master's own contribution
        auto * ml2d = (ML2DBaseProgram*)program;
        ml2d->addPartialDocfileData(docfiledata, ml2d->myFirstImg, ml2d->myLastImg);
        int s_size;
        size_t first_img, last_img;
        MPI_Status status;

        for (size_t docCounter = 1; docCounter < node->size; ++docCounter)
        {
            // receive in order
            MPI_Recv(&s_size, 1, MPI_INT, docCounter, TAG_DOCFILESIZE,
                     MPI_COMM_WORLD, &status);
            MPI_Recv(MULTIDIM_ARRAY(docfiledata), s_size,
                     MPI_DOUBLE, docCounter, TAG_DOCFILE,
                     MPI_COMM_WORLD, &status);
            divide_equally(ml2d->nr_images_global, node->size, docCounter, first_img, last_img);
            ml2d->addPartialDocfileData(docfiledata, first_img, last_img);
        }
    }
}

Member Data Documentation

created_node

template<typename T>

bool MpiML2DBase< T >::created_node

protected

Definition at line 45 of file mpi_ml_align2d.h.

node

template<typename T>

MpiNode* MpiML2DBase< T >::node

protected

Definition at line 44 of file mpi_ml_align2d.h.

program

template<typename T>

T* MpiML2DBase< T >::program

protected

Definition at line 47 of file mpi_ml_align2d.h.

---

The documentation for this class was generated from the following files:

• xmipp/libraries/parallel/mpi_ml_align2d.h
• xmipp/libraries/parallel/mpi_ml_align2d.cpp