77 строки
1.3 KiB
C
77 строки
1.3 KiB
C
/*
|
|
* Copyright (c) 2013 Sandia National Laboratories. All rights reserved.
|
|
*
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*
|
|
*/
|
|
|
|
#include "ompi_config.h"
|
|
|
|
#include <stdio.h>
|
|
#include <pmi.h>
|
|
#if WANT_PMI2_SUPPORT
|
|
#include <pmi2.h>
|
|
#endif
|
|
|
|
#include "opal/threads/tsd.h"
|
|
#include "ompi/constants.h"
|
|
#include "ompi/mca/rte/rte.h"
|
|
|
|
#include "rte_pmi.h"
|
|
|
|
|
|
static void
|
|
coll_construct(ompi_rte_collective_t *coll)
|
|
{
|
|
coll->id = 0;
|
|
coll->active = false;
|
|
}
|
|
|
|
|
|
OBJ_CLASS_INSTANCE(ompi_rte_collective_t, opal_object_t, coll_construct, NULL);
|
|
|
|
|
|
int
|
|
ompi_rte_modex(ompi_rte_collective_t *coll)
|
|
{
|
|
int len, ret;
|
|
char *kvs;
|
|
|
|
ret = PMI_KVS_Get_name_length_max(&len);
|
|
if (PMI_SUCCESS != ret) return OMPI_ERROR;
|
|
|
|
kvs = malloc(len);
|
|
if (NULL == kvs) {
|
|
return OMPI_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
|
|
ret = PMI_KVS_Get_my_name(kvs, len);
|
|
if (PMI_SUCCESS != ret) return OMPI_ERROR;
|
|
|
|
ret = PMI_KVS_Commit(kvs);
|
|
if (PMI_SUCCESS != ret) return OMPI_ERROR;
|
|
|
|
ret = PMI_Barrier();
|
|
if (PMI_SUCCESS != ret) return OMPI_ERROR;
|
|
|
|
return OMPI_SUCCESS;
|
|
}
|
|
|
|
|
|
int
|
|
ompi_rte_barrier(ompi_rte_collective_t *coll)
|
|
{
|
|
int ret;
|
|
|
|
ret = PMI_Barrier();
|
|
if (PMI_SUCCESS != ret) return OMPI_ERROR;
|
|
|
|
coll->active = false;
|
|
|
|
return OMPI_SUCCESS;
|
|
}
|