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 */