comm_mpi.cpp File Reference

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <mpi.h>
#include <quda_internal.h>
#include <comm_quda.h>

Go to the source code of this file.

Classes
struct	MsgHandle_s

Macros
#define	MPI_CHECK(mpi_call)

Functions
void	comm_init (int ndim, const int *dims, QudaCommsMap rank_from_coords, void *map_data)
int	comm_rank (void)
int	comm_size (void)
int	comm_gpuid (void)
MsgHandle *	comm_declare_send_displaced (void *buffer, const int displacement[], size_t nbytes)
MsgHandle *	comm_declare_receive_displaced (void *buffer, const int displacement[], size_t nbytes)
void	comm_free (MsgHandle *mh)
void	comm_start (MsgHandle *mh)
void	comm_wait (MsgHandle *mh)
int	comm_query (MsgHandle *mh)
void	comm_allreduce (double *data)
void	comm_allreduce_max (double *data)
void	comm_allreduce_array (double *data, size_t size)
void	comm_allreduce_int (int *data)
void	comm_broadcast (void *data, size_t nbytes)
void	comm_barrier (void)
void	comm_abort (int status)

Macro Definition Documentation

#define MPI_CHECK

(

mpi_call

)

Value:

do {                    \
  int status = mpi_call;                            \
  if (status != MPI_SUCCESS) {                      \
    char err_string[128];                           \
    int err_len;                                    \
    MPI_Error_string(status, err_string, &err_len); \
    err_string[127] = '\0';                         \
    errorQuda("(MPI) %s", err_string);              \
  }                                                 \
} while (0)

Definition at line 10 of file comm_mpi.cpp.

Function Documentation

void comm_abort

(

int

status

)

Definition at line 204 of file comm_mpi.cpp.

void comm_allreduce

(

double *

data

)

Definition at line 159 of file comm_mpi.cpp.

void comm_allreduce_array	(	double *	data,
		size_t	size
	)

Definition at line 175 of file comm_mpi.cpp.

void comm_allreduce_int

(

int *

data

)

Definition at line 183 of file comm_mpi.cpp.

void comm_allreduce_max

(

double *

data

)

Definition at line 167 of file comm_mpi.cpp.

void comm_barrier

(

void

)

Definition at line 198 of file comm_mpi.cpp.

void comm_broadcast	(	void *	data,
		size_t	nbytes
	)

broadcast from rank 0

Definition at line 192 of file comm_mpi.cpp.

MsgHandle* comm_declare_receive_displaced	(	void *	buffer,
		const int	displacement[],
		size_t	nbytes
	)

Declare a message handle for receiving from a node displaced in (x,y,z,t) according to "displacement"

Definition at line 119 of file comm_mpi.cpp.

MsgHandle* comm_declare_send_displaced	(	void *	buffer,
		const int	displacement[],
		size_t	nbytes
	)

Declare a message handle for sending to a node displaced in (x,y,z,t) according to "displacement"

Definition at line 103 of file comm_mpi.cpp.

void comm_free

(

MsgHandle *

mh

)

Definition at line 132 of file comm_mpi.cpp.

int comm_gpuid

(

void

)

Definition at line 94 of file comm_mpi.cpp.

void comm_init	(	int	ndim,
		const int *	dims,
		QudaCommsMap	rank_from_coords,
		void *	map_data
	)

Definition at line 32 of file comm_mpi.cpp.

int comm_query

(

MsgHandle *

mh

)

Definition at line 150 of file comm_mpi.cpp.

int comm_rank

(

void

)

Definition at line 82 of file comm_mpi.cpp.

int comm_size

(

void

)

Definition at line 88 of file comm_mpi.cpp.

void comm_start

(

MsgHandle *

mh

)

Definition at line 138 of file comm_mpi.cpp.

void comm_wait

(

MsgHandle *

mh

)

Definition at line 144 of file comm_mpi.cpp.