This source file includes following definitions.
- mca_coll_monitoring_alltoallw
- mca_coll_monitoring_ialltoallw
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_alltoallw(const void *sbuf, const int *scounts,
17 const int *sdisps,
18 struct ompi_datatype_t * const *sdtypes,
19 void *rbuf, const int *rcounts,
20 const int *rdisps,
21 struct ompi_datatype_t * const *rdtypes,
22 struct ompi_communicator_t *comm,
23 mca_coll_base_module_t *module)
24 {
25 mca_coll_monitoring_module_t*monitoring_module = (mca_coll_monitoring_module_t*) module;
26 size_t type_size, data_size, data_size_aggreg = 0;
27 const int comm_size = ompi_comm_size(comm);
28 const int my_rank = ompi_comm_rank(comm);
29 int i, rank;
30 for( i = 0; i < comm_size; ++i ) {
31 if( my_rank == i ) continue;
32 ompi_datatype_type_size(sdtypes[i], &type_size);
33 data_size = scounts[i] * type_size;
34
35
36
37
38 if( OPAL_SUCCESS == mca_common_monitoring_get_world_rank(i, comm->c_remote_group, &rank) ) {
39 mca_common_monitoring_record_coll(rank, data_size);
40 data_size_aggreg += data_size;
41 }
42 }
43 mca_common_monitoring_coll_a2a(data_size_aggreg, monitoring_module->data);
44 return monitoring_module->real.coll_alltoallw(sbuf, scounts, sdisps, sdtypes, rbuf, rcounts, rdisps, rdtypes, comm, monitoring_module->real.coll_alltoallw_module);
45 }
46
47 int mca_coll_monitoring_ialltoallw(const void *sbuf, const int *scounts,
48 const int *sdisps,
49 struct ompi_datatype_t * const *sdtypes,
50 void *rbuf, const int *rcounts,
51 const int *rdisps,
52 struct ompi_datatype_t * const *rdtypes,
53 struct ompi_communicator_t *comm,
54 ompi_request_t ** request,
55 mca_coll_base_module_t *module)
56 {
57 mca_coll_monitoring_module_t*monitoring_module = (mca_coll_monitoring_module_t*) module;
58 size_t type_size, data_size, data_size_aggreg = 0;
59 const int comm_size = ompi_comm_size(comm);
60 const int my_rank = ompi_comm_rank(comm);
61 int i, rank;
62 for( i = 0; i < comm_size; ++i ) {
63 if( my_rank == i ) continue;
64 ompi_datatype_type_size(sdtypes[i], &type_size);
65 data_size = scounts[i] * type_size;
66
67
68
69
70 if( OPAL_SUCCESS == mca_common_monitoring_get_world_rank(i, comm->c_remote_group, &rank) ) {
71 mca_common_monitoring_record_coll(rank, data_size);
72 data_size_aggreg += data_size;
73 }
74 }
75 mca_common_monitoring_coll_a2a(data_size_aggreg, monitoring_module->data);
76 return monitoring_module->real.coll_ialltoallw(sbuf, scounts, sdisps, sdtypes, rbuf, rcounts, rdisps, rdtypes, comm, request, monitoring_module->real.coll_ialltoallw_module);
77 }