99 строки
4.0 KiB
C
99 строки
4.0 KiB
C
|
/*
|
||
|
|
* Copyright (c) 2004-2006 The Trustees of Indiana University and Indiana
|
||
|
|
* University Research and Technology
|
||
|
|
* Corporation. All rights reserved.
|
||
|
|
* Copyright (c) 2004-2010 The University of Tennessee and The University
|
||
|
|
* of Tennessee Research Foundation. All rights
|
||
|
|
* reserved.
|
||
|
|
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
|
||
|
|
* University of Stuttgart. All rights reserved.
|
||
|
|
* Copyright (c) 2004-2006 The Regents of the University of California.
|
||
|
|
* All rights reserved.
|
||
|
|
* Copyright (c) 2006 QLogic Corporation. All rights reserved.
|
||
|
|
* Copyright (c) 2006-2017 Los Alamos National Security, LLC. All rights
|
||
|
|
* reserved.
|
||
|
|
* Copyright (c) 2013-2015 Intel, Inc. All rights reserved
|
||
|
|
* $COPYRIGHT$
|
||
|
|
*
|
||
|
|
* Additional copyrights may follow
|
||
|
|
*
|
||
|
|
* $HEADER$
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "ompi_config.h"
|
||
|
|
#include "mtl_psm2.h"
|
||
|
|
#include "mtl_psm2_types.h"
|
||
|
|
#include "psm2.h"
|
||
|
|
#include "ompi/communicator/communicator.h"
|
||
|
|
#include "ompi/message/message.h"
|
||
|
|
|
||
|
|
#include "opal/mca/base/mca_base_pvar.h"
|
||
|
|
|
||
|
|
struct ompi_mtl_psm2_name_descs
|
||
|
|
{
|
||
|
|
char *name;
|
||
|
|
char *desc;
|
||
|
|
ptrdiff_t offset;
|
||
|
|
};
|
||
|
|
|
||
|
|
const struct ompi_mtl_psm2_name_descs name_descs[PSM2_MQ_NUM_STATS] =
|
||
|
|
{
|
||
|
|
{ "rx_user_bytes", "Bytes received into a matched user buffer",
|
||
|
|
offsetof(struct psm2_mq_stats, rx_user_bytes) },
|
||
|
|
{ "rx_user_num", "Messages received into a matched user buffer",
|
||
|
|
offsetof(struct psm2_mq_stats, rx_user_num) },
|
||
|
|
{ "rx_sys_bytes", "Bytes received into an unmatched system buffer",
|
||
|
|
offsetof(struct psm2_mq_stats, rx_sys_bytes) },
|
||
|
|
{ "rx_sys_num", "Messages received into an unmatched system buffer",
|
||
|
|
offsetof(struct psm2_mq_stats, rx_sys_num) },
|
||
|
|
{ "tx_num", "Total Messages transmitted (shm and hfi)",
|
||
|
|
offsetof(struct psm2_mq_stats, tx_num) },
|
||
|
|
{ "tx_eager_num", "Messages transmitted eagerly",
|
||
|
|
offsetof(struct psm2_mq_stats, tx_eager_num) },
|
||
|
|
{ "tx_eager_bytes", "Bytes transmitted eagerl",
|
||
|
|
offsetof(struct psm2_mq_stats, tx_eager_bytes) },
|
||
|
|
{ "tx_rndv_num", "Messages transmitted using expected TID mechanism",
|
||
|
|
offsetof(struct psm2_mq_stats, tx_rndv_num) },
|
||
|
|
{ "tx_rndv_bytes", "Bytes transmitted using expected TID mechanism",
|
||
|
|
offsetof(struct psm2_mq_stats, tx_rndv_bytes) },
|
||
|
|
{ "tx_shm_num", "Messages transmitted (shm only)",
|
||
|
|
offsetof(struct psm2_mq_stats, tx_shm_num) },
|
||
|
|
{ "rx_shm_num", "Messages received through shm",
|
||
|
|
offsetof(struct psm2_mq_stats, rx_shm_num) },
|
||
|
|
{ "rx_sysbuf_num", "Number of system buffers allocated",
|
||
|
|
offsetof(struct psm2_mq_stats, rx_sysbuf_num) },
|
||
|
|
{ "rx_sysbuf_bytes", "Bytes allocated for system buffers",
|
||
|
|
offsetof(struct psm2_mq_stats, rx_sysbuf_bytes) },
|
||
|
|
};
|
||
|
|
|
||
|
|
static int mca_mtl_psm2_get_stats(const mca_base_pvar_t *pvar, void *value, void *obj)
|
||
|
|
{
|
||
|
|
psm2_mq_stats_t stats;
|
||
|
|
int index = (int)(intptr_t) pvar->ctx;
|
||
|
|
|
||
|
|
psm2_mq_get_stats(ompi_mtl_psm2.mq, &stats);
|
||
|
|
|
||
|
|
*(uint64_t *)value = *(uint64_t *)((uint8_t *)&stats + name_descs[index].offset);
|
||
|
|
|
||
|
|
return OMPI_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
int ompi_mtl_psm2_register_pvars(void)
|
||
|
|
{
|
||
|
|
int i;
|
||
|
|
|
||
|
|
/* PSM2 MQ performance variables */
|
||
|
|
for (i = 0 ; i < PSM2_MQ_NUM_STATS; ++i) {
|
||
|
|
(void) mca_base_component_pvar_register (&mca_mtl_psm2_component.super.mtl_version,
|
||
|
|
name_descs[i].name, name_descs[i].desc,
|
||
|
|
OPAL_INFO_LVL_4, MCA_BASE_PVAR_CLASS_COUNTER,
|
||
|
|
MCA_BASE_VAR_TYPE_UNSIGNED_LONG,
|
||
|
|
NULL, MCA_BASE_VAR_BIND_NO_OBJECT,
|
||
|
|
MCA_BASE_PVAR_FLAG_READONLY | MCA_BASE_PVAR_FLAG_CONTINUOUS,
|
||
|
|
mca_mtl_psm2_get_stats, NULL, NULL,
|
||
|
|
(void *) (intptr_t) i);
|
||
|
|
}
|
||
|
|
return OMPI_SUCCESS;
|
||
|
|
}
|