cf6137b530
Bring Slurm PMI-1 component online Bring the s2 component online Little cleanup - let the various PMIx modules set the process name during init, and then just raise it up to the ORTE level. Required as the different PMI environments all pass the jobid in different ways. Bring the OMPI pubsub/pmi component online Get comm_spawn working again Ensure we always provide a cpuset, even if it is NULL pmix/cray: adjust cray pmix component for pmix Make changes so cray pmix can work within the integrated ompi/pmix framework. Bring singletons back online. Implement the comm_spawn operation using pmix - not tested yet Cleanup comm_spawn - procs now starting, error in connect_accept Complete integration
128 строки
4.2 KiB
C
128 строки
4.2 KiB
C
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil -*- */
|
|
/*
|
|
* Copyright (c) 2004-2007 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2005 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-2005 The Regents of the University of California.
|
|
* All rights reserved.
|
|
* Copyright (c) 2012-2013 Los Alamos National Security, LLC. All rights
|
|
* reserved.
|
|
* Copyright (c) 2015 Intel, Inc. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
#include "ompi_config.h"
|
|
#include <stdio.h>
|
|
|
|
#include "opal/class/opal_list.h"
|
|
#include "opal/mca/pmix/pmix.h"
|
|
#include "opal/util/argv.h"
|
|
|
|
#include "ompi/mpi/c/bindings.h"
|
|
#include "ompi/runtime/params.h"
|
|
#include "ompi/communicator/communicator.h"
|
|
#include "ompi/errhandler/errhandler.h"
|
|
#include "ompi/info/info.h"
|
|
|
|
#if OPAL_HAVE_WEAK_SYMBOLS && OMPI_PROFILING_DEFINES
|
|
#pragma weak MPI_Unpublish_name = PMPI_Unpublish_name
|
|
#endif
|
|
|
|
#if OMPI_PROFILING_DEFINES
|
|
#include "ompi/mpi/c/profile/defines.h"
|
|
#endif
|
|
|
|
static const char FUNC_NAME[] = "MPI_Unpublish_name";
|
|
|
|
|
|
int MPI_Unpublish_name(const char *service_name, MPI_Info info,
|
|
const char *port_name)
|
|
{
|
|
int rc;
|
|
char range[OPAL_MAX_INFO_VAL];
|
|
int flag=0;
|
|
opal_pmix_data_range_t rng;
|
|
bool range_given = false;
|
|
char **keys = NULL;
|
|
|
|
if ( MPI_PARAM_CHECK ) {
|
|
OMPI_ERR_INIT_FINALIZE(FUNC_NAME);
|
|
|
|
if ( NULL == port_name ) {
|
|
return OMPI_ERRHANDLER_INVOKE(MPI_COMM_WORLD, MPI_ERR_ARG,
|
|
FUNC_NAME);
|
|
}
|
|
if ( NULL == service_name ) {
|
|
return OMPI_ERRHANDLER_INVOKE(MPI_COMM_WORLD, MPI_ERR_ARG,
|
|
FUNC_NAME);
|
|
}
|
|
if (NULL == info || ompi_info_is_freed(info)) {
|
|
return OMPI_ERRHANDLER_INVOKE(MPI_COMM_WORLD, MPI_ERR_INFO,
|
|
FUNC_NAME);
|
|
}
|
|
}
|
|
|
|
OPAL_CR_ENTER_LIBRARY();
|
|
|
|
/* OMPI supports info keys to pass the range to
|
|
* be searched for the given key */
|
|
if (MPI_INFO_NULL != info) {
|
|
ompi_info_get (info, "range", sizeof(range) - 1, range, &flag);
|
|
if (flag) {
|
|
range_given = true;
|
|
if (0 == strcmp(range, "nspace")) {
|
|
rng = OPAL_PMIX_NAMESPACE; // share only with procs in same nspace
|
|
} else if (0 == strcmp(range, "session")) {
|
|
rng = OPAL_PMIX_SESSION; // share only with procs in same session
|
|
} else {
|
|
/* unrecognized range */
|
|
return OMPI_ERRHANDLER_INVOKE(MPI_COMM_WORLD, MPI_ERR_ARG,
|
|
FUNC_NAME);
|
|
}
|
|
}
|
|
}
|
|
if (!range_given) {
|
|
/* default to nspace */
|
|
rng = OPAL_PMIX_NAMESPACE;
|
|
}
|
|
|
|
/* unpublish the service_name */
|
|
opal_argv_append_nosize(&keys, service_name);
|
|
|
|
rc = opal_pmix.unpublish(rng, keys);
|
|
opal_argv_free(keys);
|
|
|
|
if ( OPAL_SUCCESS != rc ) {
|
|
if (OPAL_ERR_NOT_FOUND == rc) {
|
|
/* service couldn't be found */
|
|
OPAL_CR_EXIT_LIBRARY();
|
|
return OMPI_ERRHANDLER_INVOKE(MPI_COMM_WORLD, MPI_ERR_SERVICE,
|
|
FUNC_NAME);
|
|
}
|
|
if (OPAL_ERR_PERM == rc) {
|
|
/* this process didn't own the specified service */
|
|
OPAL_CR_EXIT_LIBRARY();
|
|
return OMPI_ERRHANDLER_INVOKE(MPI_COMM_WORLD, MPI_ERR_ACCESS,
|
|
FUNC_NAME);
|
|
}
|
|
|
|
/* none of the MPI-specific errors occurred - must be some
|
|
* kind of internal error
|
|
*/
|
|
OPAL_CR_EXIT_LIBRARY();
|
|
return OMPI_ERRHANDLER_INVOKE(MPI_COMM_WORLD, MPI_ERR_INTERN,
|
|
FUNC_NAME);
|
|
}
|
|
|
|
OPAL_CR_EXIT_LIBRARY();
|
|
return MPI_SUCCESS;
|
|
}
|