src/smpi/include/smpi_comm.hpp

   1 /* Copyright (c) 2010-2017. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #ifndef SMPI_COMM_HPP_INCLUDED
   7 #define SMPI_COMM_HPP_INCLUDED
   8
   9 #include <list>
  10 #include "smpi_keyvals.hpp"
  11 #include "smpi_group.hpp"
  12 #include "smpi_topo.hpp"
  13
  14 namespace simgrid{
  15 namespace smpi{
  16
  17 class Comm : public F2C, public Keyval{
  18
  19   private:
  20     MPI_Group group_;
  21     SMPI_Topo_type topoType_;
  22     MPI_Topology topo_; // to be replaced by an union
  23     int refcount_;
  24     MPI_Comm leaders_comm_;//inter-node communicator
  25     MPI_Comm intra_comm_;//intra-node communicator . For MPI_COMM_WORLD this can't be used, as var is global.
  26     //use an intracomm stored in the process data instead
  27     int* leaders_map_; //who is the leader of each process
  28     int is_uniform_;
  29     int* non_uniform_map_; //set if smp nodes have a different number of processes allocated
  30     int is_blocked_;// are ranks allocated on the same smp node contiguous ?
  31
  32     std::list<MPI_Win> rma_wins_; // attached windows for synchronization.
  33
  34   public:
  35     static std::unordered_map<int, smpi_key_elem> keyvals_;
  36     static int keyval_id_;
  37
  38     Comm() = default;
  39     Comm(MPI_Group group, MPI_Topology topo);
  40     int dup(MPI_Comm* newcomm);
  41     MPI_Group group();
  42     MPI_Topology topo();
  43     int size();
  44     int rank();
  45     void get_name (char* name, int* len);
  46     void set_leaders_comm(MPI_Comm leaders);
  47     void set_intra_comm(MPI_Comm leaders);
  48     int* get_non_uniform_map();
  49     int* get_leaders_map();
  50     MPI_Comm get_leaders_comm();
  51     MPI_Comm get_intra_comm();
  52     int is_uniform();
  53     int is_blocked();
  54     MPI_Comm split(int color, int key);
  55     void cleanup_smp();
  56     void ref();
  57     static void unref(MPI_Comm comm);
  58     static void destroy(MPI_Comm comm);
  59     void init_smp();
  60
  61     int add_f() override;
  62     static void free_f(int id);
  63     static Comm* f2c(int);
  64
  65     static int keyval_create(MPI_Comm_copy_attr_function* copy_fn, MPI_Comm_delete_attr_function* delete_fn, int* keyval, void* extra_state);
  66     static int keyval_free(int* keyval);
  67     static void keyval_cleanup();
  68
  69     void add_rma_win(MPI_Win win);
  70     void remove_rma_win(MPI_Win win);
  71     void finish_rma_calls();
  72
  73 };
  74
  75 }
  76 }
  77
  78
  79 #endif