communicator.h

Go to the documentation of this file.

 
#ifndef COMMUNICATOR_INCLUDED
#define COMMUNICATOR_INCLUDED
 
#include "configure.h"
#include "bridge_complex.h"
#include "bridge_defs.h"
 
#include <cstdio>
#include <cstdlib>
#include <cstddef>
#include <vector>
#include <string>
using std::string;
 
class Channel;
 
 
// forward declaration
//class Channel;
 
class Communicator {
 public:
 
  static int init(int *pargc, char ***pargv);
 
 
  static int finalize();
 
 
  static void abort();
 
 
  static int setup(int ninstance = 1);
 
  static int setup(const std::vector<int>& lattice_size,
                   std::vector<int>& grid_size,
                   int ninstance = 1);
 
// info about rank
  static bool is_primary();        
  static bool is_primary_master(); 
 
  static int self();               
 
  static int nodeid() { return self(); }  
  static int size();   
 
#ifdef ENABLE_MULTI_INSTANCE
  static int self_global(); 
  static int world_id();    
#endif
 
// layout
  static int ipe(const int dir);                          
  static int npe(const int dir);                          
 
  static int grid_rank(int *rank, const int *grid_coord); 
  static int grid_coord(int *grid_coord, const int rank); 
  static int grid_dims(int *grid_dims);                   
 
// synchronize
  static int sync();   
 
// synchronize (w/o busy wait)
  static int sync_usleep();   
 
#ifdef ENABLE_MULTI_INSTANCE
  static int sync_global();   
#endif
 
// data transfer
  static int broadcast(int count, dcomplex *data, int sender);                                            
  static int broadcast(int count, double *data, int sender);                                              
  static int broadcast(int count, float *data, int sender);                                               
  static int broadcast(int count, int *data, int sender);                                                 
  static int broadcast(int count, string& data, int sender);                                              
 
  static int exchange(int count, dcomplex *recv_buf, dcomplex *send_buf, int idir, int ipm, int tag);     
  static int exchange(int count, double *recv_buf, double *send_buf, int idir, int ipm, int tag);         
  static int exchange(int count, float *recv_buf, float *send_buf, int idir, int ipm, int tag);           
  static int exchange(int count, int *recv_buf, int *send_buf, int idir, int ipm, int tag);               
 
  static int send_1to1(int count, dcomplex *recv_buf, dcomplex *send_buf, int p_to, int p_from, int tag); 
  static int send_1to1(int count, double *recv_buf, double *send_buf, int p_to, int p_from, int tag);     
  static int send_1to1(int count, float *recv_buf, float *send_buf, int p_to, int p_from, int tag);       
  static int send_1to1(int count, int *recv_buf, int *send_buf, int p_to, int p_from, int tag);           
 
  static int reduce_sum(int count, dcomplex *recv_buf, dcomplex *send_buf, int pattern = 0);              
  static int reduce_sum(int count, double *recv_buf, double *send_buf, int pattern     = 0);              
  static int reduce_sum(int count, float *recv_buf, float *send_buf, int pattern       = 0);              
  static int reduce_sum(int count, int *recv_buf, int *send_buf, int pattern           = 0);              
 
  static int reduce_max(int count, double *recv_buf, double *send_buf, int pattern = 0);                  
  static int reduce_max(int count, float *recv_buf, float *send_buf, int pattern   = 0);                  
  static int reduce_max(int count, int *recv_buf, int *send_buf, int pattern       = 0);                  
 
  static int reduce_min(int count, double *recv_buf, double *send_buf, int pattern = 0);                  
  static int reduce_min(int count, float *recv_buf, float *send_buf, int pattern   = 0);                  
  static int reduce_min(int count, int *recv_buf, int *send_buf, int pattern       = 0);                  
 
  static dcomplex reduce_sum(dcomplex);                                                                   
  //- NB. no reduce_{max,min} for dcomplex
  // static dcomplex reduce_max(dcomplex);
  // static dcomplex reduce_min(dcomplex);
 
  static double reduce_sum(double);                                                                   
  static double reduce_max(double);                                                                   
  static double reduce_min(double);                                                                   
 
  static float reduce_sum(float);                                                                     
  static float reduce_max(float);                                                                     
  static float reduce_min(float);                                                                     
 
  static double get_time();                                                                           
 
  // async communication
  static Channel *send_init(int count, int idir, int ipm);
  static Channel *recv_init(int count, int idir, int ipm);
 
  // async communication with given buffer [2017.09.02 H.Matsufuru]
  static Channel *send_init(int count, int idir, int ipm, void *buf);
  static Channel *recv_init(int count, int idir, int ipm, void *buf);
 
// debug
  static int status();
 
 
  class Base {
   public:
    static int broadcast(size_t size, void *data, int sender);
    static int exchange(size_t size, void *recv_buf, void *send_buf, int idir, int ipm, int tag);
 
    static int send_1to1(size_t size, void *recv_buf, void *send_buf, int send_to, int recv_from, int tag);
  };
 
 private:
 
 
  Communicator() {}
  Communicator(const Communicator&) {}
  Communicator& operator=(const Communicator&);
 
  ~Communicator() {}
};
#endif /* COMMUNICATOR_INCLUDED */