xmp_bcast.c File Reference

#include <stdio.h>
#include <stdarg.h>
#include "mpi.h"
#include "xmp_internal.h"
#include "xmp_math_function.h"

Include dependency graph for xmp_bcast.c:

Macros
#define	MPI_PORTABLE_PLATFORM_H

Functions
void	_XMP_bcast_NODES_ENTIRE_OMITTED (_XMP_nodes_t *bcast_nodes, void *addr, int count, size_t datatype_size)
void	_XMP_bcast_NODES_ENTIRE_NODES (_XMP_nodes_t *bcast_nodes, void *addr, int count, size_t datatype_size, _XMP_nodes_t *from_nodes,...)
void	_XMP_bcast_NODES_ENTIRE_NODES_V (_XMP_nodes_t *bcast_nodes, void *addr, int count, size_t datatype_size, _XMP_nodes_t *from_nodes, va_list args)

Macro Definition Documentation

MPI_PORTABLE_PLATFORM_H

#define MPI_PORTABLE_PLATFORM_H

Function Documentation

_XMP_bcast_NODES_ENTIRE_NODES()

void _XMP_bcast_NODES_ENTIRE_NODES	(	_XMP_nodes_t *	bcast_nodes,
		void *	addr,
		int	count,
		size_t	datatype_size,
		_XMP_nodes_t *	from_nodes,
			...
	)

                                                                  {
  va_list args;
  va_start(args, from_nodes);
 
  _XMP_bcast_NODES_ENTIRE_NODES_V(bcast_nodes, addr, count, datatype_size, from_nodes, args);
 
  va_end(args);
}

Here is the call graph for this function:

_XMP_bcast_NODES_ENTIRE_NODES_V()

void _XMP_bcast_NODES_ENTIRE_NODES_V	(	_XMP_nodes_t *	bcast_nodes,
		void *	addr,
		int	count,
		size_t	datatype_size,
		_XMP_nodes_t *	from_nodes,
		va_list	args
	)

                                                                             {
  _XMP_RETURN_IF_SINGLE;
 
  if (!bcast_nodes->is_member) {
    return;
  }
 
  if (!from_nodes->is_member) {
    _XMP_fatal("broadcast failed, cannot find the source node");
  }
 
  // calc source nodes number
  int root = 0;
  int acc_nodes_size = 1;
  int from_dim = from_nodes->dim;
  int from_lower, from_upper, from_stride;
  _XMP_nodes_inherit_info_t  *inherit_info = bcast_nodes->inherit_info;
 
  if(inherit_info == NULL){
    for (int i = 0; i < from_dim; i++) {
      int size = from_nodes->info[i].size;
      if(inherit_info != NULL){
        if(inherit_info[i].shrink == true)
          continue;
        size = inherit_info[i].upper - inherit_info[i].lower + 1;
        if(size == 0) continue;
      }
      int rank = from_nodes->info[i].rank;
 
      if (va_arg(args, int) == 1) {
        root += (acc_nodes_size * rank);
      }
      else {
        from_lower = va_arg(args, int) - 1;
        from_upper = va_arg(args, int) - 1;
        from_stride = va_arg(args, int);
        
        // check <from-ref>
        if (_XMP_M_COUNT_TRIPLETi(from_lower, from_upper, from_stride) != 1) {
          _XMP_fatal("multiple source nodes indicated in bcast directive");
        }
 
        root += (acc_nodes_size * (from_lower));
      }
      
      acc_nodes_size *= size;
    }
  }
  else{
    int inherit_node_dim = bcast_nodes->inherit_nodes->dim;
 
    for (int i = 0; i < inherit_node_dim; i++) {
 
      if(inherit_info[i].shrink) // skip i
        continue;
 
      int size = inherit_info[i].upper - inherit_info[i].lower + 1;
      
      if(size == 0) {  // skip arguments
        va_arg(args, int);   // is_astrisk 
        va_arg(args, int);   // from_lower
        va_arg(args, int);   // from_upper
        va_arg(args, int);   // from_stride
        continue;
      }
 
      int is_astrisk = va_arg(args, int);
      if (is_astrisk == 1){
        int rank = from_nodes->info[i].rank;
        root += (acc_nodes_size * rank);
      }
      else {
        from_lower = va_arg(args, int) - 1;
        from_upper = va_arg(args, int) - 1;
        va_arg(args, int); // skip from_stride
 
        // check <from-ref> 
        if(from_lower != from_upper)
          _XMP_fatal("multiple source nodes indicated in bcast directive");
 
        root += (acc_nodes_size * (from_lower - inherit_info[i].lower));
      }
      
      acc_nodes_size *= size;
    }
  }
 
#ifdef _XMP_MPI3
  if(xmp_is_async()){
    _XMP_async_comm_t *async = _XMP_get_current_async();
    MPI_Ibcast(addr, count*datatype_size, MPI_BYTE, root,
               *((MPI_Comm *)bcast_nodes->comm), &async->reqs[async->nreqs]);
    async->nreqs++;
  }
  else
#endif
  MPI_Bcast(addr, count*datatype_size, MPI_BYTE, root,
            *((MPI_Comm *)bcast_nodes->comm));
 
  /* // setup type */
  /* MPI_Datatype mpi_datatype; */
  /* MPI_Type_contiguous(datatype_size, MPI_BYTE, &mpi_datatype); */
  /* MPI_Type_commit(&mpi_datatype); */
 
  /* // bcast */
  /* MPI_Bcast(addr, count, mpi_datatype, root, *((MPI_Comm *)bcast_nodes->comm)); */
 
  /* MPI_Type_free(&mpi_datatype); */
}

Here is the call graph for this function:

Here is the caller graph for this function:

_XMP_bcast_NODES_ENTIRE_OMITTED()

void _XMP_bcast_NODES_ENTIRE_OMITTED	(	_XMP_nodes_t *	bcast_nodes,
		void *	addr,
		int	count,
		size_t	datatype_size
	)

                                                                                                            {
  _XMP_RETURN_IF_SINGLE;
 
  if(!bcast_nodes->is_member) return;
 
#ifdef _XMP_MPI3
  if(xmp_is_async()){
    _XMP_async_comm_t *async = _XMP_get_current_async();
    MPI_Ibcast(addr, count*datatype_size, MPI_BYTE, _XMP_N_DEFAULT_ROOT_RANK,
               *((MPI_Comm *)bcast_nodes->comm), &async->reqs[async->nreqs]);
    async->nreqs++;
  }
  else
#endif
    MPI_Bcast(addr, count*datatype_size, MPI_BYTE, _XMP_N_DEFAULT_ROOT_RANK,
              *((MPI_Comm *)bcast_nodes->comm));
}

Here is the call graph for this function: