This source file includes following definitions.
- mca_coll_monitoring_scatterv
- mca_coll_monitoring_iscatterv
   1 
   2 
   3 
   4 
   5 
   6 
   7 
   8 
   9 
  10 #include <ompi_config.h>
  11 #include <ompi/request/request.h>
  12 #include <ompi/datatype/ompi_datatype.h>
  13 #include <ompi/communicator/communicator.h>
  14 #include "coll_monitoring.h"
  15 
  16 int mca_coll_monitoring_scatterv(const void *sbuf, const int *scounts, const int *disps,
  17                                  struct ompi_datatype_t *sdtype,
  18                                  void* rbuf, int rcount, struct ompi_datatype_t *rdtype,
  19                                  int root, struct ompi_communicator_t *comm,
  20                                  mca_coll_base_module_t *module)
  21 {
  22     mca_coll_monitoring_module_t*monitoring_module = (mca_coll_monitoring_module_t*) module;
  23     const int my_rank = ompi_comm_rank(comm);
  24     if( root == my_rank ) {
  25         size_t type_size, data_size, data_size_aggreg = 0;
  26         const int comm_size = ompi_comm_size(comm);
  27         int i, rank;
  28         ompi_datatype_type_size(sdtype, &type_size);
  29         for( i = 0; i < comm_size; ++i ) {
  30             data_size = scounts[i] * type_size;
  31             
  32 
  33 
  34 
  35             if( OPAL_SUCCESS == mca_common_monitoring_get_world_rank(i, comm->c_remote_group, &rank) ) {
  36                 mca_common_monitoring_record_coll(rank, data_size);
  37                 data_size_aggreg += data_size;
  38             }
  39         }
  40         mca_common_monitoring_coll_o2a(data_size_aggreg, monitoring_module->data);
  41     }
  42     return monitoring_module->real.coll_scatterv(sbuf, scounts, disps, sdtype, rbuf, rcount, rdtype, root, comm, monitoring_module->real.coll_scatterv_module);
  43 }
  44 
  45 int mca_coll_monitoring_iscatterv(const void *sbuf, const int *scounts, const int *disps,
  46                                   struct ompi_datatype_t *sdtype,
  47                                   void *rbuf, int rcount, struct ompi_datatype_t *rdtype,
  48                                   int root, struct ompi_communicator_t *comm,
  49                                   ompi_request_t ** request,
  50                                   mca_coll_base_module_t *module)
  51 {
  52     mca_coll_monitoring_module_t*monitoring_module = (mca_coll_monitoring_module_t*) module;
  53     const int my_rank = ompi_comm_rank(comm);
  54     if( root == my_rank ) {
  55         size_t type_size, data_size, data_size_aggreg = 0;
  56         const int comm_size = ompi_comm_size(comm);
  57         int i, rank;
  58         ompi_datatype_type_size(sdtype, &type_size);
  59         for( i = 0; i < comm_size; ++i ) {
  60             data_size = scounts[i] * type_size;
  61             
  62 
  63 
  64 
  65             if( OPAL_SUCCESS == mca_common_monitoring_get_world_rank(i, comm->c_remote_group, &rank) ) {
  66                 mca_common_monitoring_record_coll(rank, data_size);
  67                 data_size_aggreg += data_size;
  68             }
  69         }
  70         mca_common_monitoring_coll_o2a(data_size_aggreg, monitoring_module->data);
  71     }
  72     return monitoring_module->real.coll_iscatterv(sbuf, scounts, disps, sdtype, rbuf, rcount, rdtype, root, comm, request, monitoring_module->real.coll_iscatterv_module);
  73 }