/* * Copyright (c) 2016-2018 Inria. All rights reserved. * Copyright (c) 2019 Research Organization for Information Science * and Technology (RIST). All rights reserved. * $COPYRIGHT$ * * Additional copyrights may follow * * $HEADER$ */ #include "ompi_config.h" #include "ompi/request/request.h" #include "ompi/datatype/ompi_datatype.h" #include "ompi/communicator/communicator.h" #include "coll_monitoring.h" int mca_coll_monitoring_scatter(const void *sbuf, int scount, struct ompi_datatype_t *sdtype, void *rbuf, int rcount, struct ompi_datatype_t *rdtype, int root, struct ompi_communicator_t *comm, mca_coll_base_module_t *module) { mca_coll_monitoring_module_t*monitoring_module = (mca_coll_monitoring_module_t*) module; const int my_rank = ompi_comm_rank(comm); if( root == my_rank ) { size_t type_size, data_size; const int comm_size = ompi_comm_size(comm); int i, rank; ompi_datatype_type_size(sdtype, &type_size); data_size = scount * type_size; for( i = 0; i < comm_size; ++i ) { if( my_rank == i ) continue; /* No communication for self */ /** * If this fails the destination is not part of my MPI_COM_WORLD * Lookup its name in the rank hashtable to get its MPI_COMM_WORLD rank */ if( OPAL_SUCCESS == mca_common_monitoring_get_world_rank(i, comm->c_remote_group, &rank) ) { mca_common_monitoring_record_coll(rank, data_size); } } mca_common_monitoring_coll_o2a(data_size * (comm_size - 1), monitoring_module->data); } return monitoring_module->real.coll_scatter(sbuf, scount, sdtype, rbuf, rcount, rdtype, root, comm, monitoring_module->real.coll_scatter_module); } int mca_coll_monitoring_iscatter(const void *sbuf, int scount, struct ompi_datatype_t *sdtype, void *rbuf, int rcount, struct ompi_datatype_t *rdtype, int root, struct ompi_communicator_t *comm, ompi_request_t ** request, mca_coll_base_module_t *module) { mca_coll_monitoring_module_t*monitoring_module = (mca_coll_monitoring_module_t*) module; const int my_rank = ompi_comm_rank(comm); if( root == my_rank ) { size_t type_size, data_size; const int comm_size = ompi_comm_size(comm); int i, rank; ompi_datatype_type_size(sdtype, &type_size); data_size = scount * type_size; for( i = 0; i < comm_size; ++i ) { if( my_rank == i ) continue; /* No communication for self */ /** * If this fails the destination is not part of my MPI_COM_WORLD * Lookup its name in the rank hashtable to get its MPI_COMM_WORLD rank */ if( OPAL_SUCCESS == mca_common_monitoring_get_world_rank(i, comm->c_remote_group, &rank) ) { mca_common_monitoring_record_coll(rank, data_size); } } mca_common_monitoring_coll_o2a(data_size * (comm_size - 1), monitoring_module->data); } return monitoring_module->real.coll_iscatter(sbuf, scount, sdtype, rbuf, rcount, rdtype, root, comm, request, monitoring_module->real.coll_iscatter_module); }