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
110 строки
3.6 KiB
C
110 строки
3.6 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-2008 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) 2006 Cisco Systems, Inc. All rights reserved.
|
|
* Copyright (c) 2008 University of Houston, Inc. All rights reserved.
|
|
* Copyright (c) 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 "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"
|
|
#include "ompi/dpm/dpm.h"
|
|
#include "ompi/memchecker.h"
|
|
|
|
#if OPAL_HAVE_WEAK_SYMBOLS && OMPI_PROFILING_DEFINES
|
|
#pragma weak MPI_Comm_accept = PMPI_Comm_accept
|
|
#endif
|
|
|
|
#if OMPI_PROFILING_DEFINES
|
|
#include "ompi/mpi/c/profile/defines.h"
|
|
#endif
|
|
|
|
static const char FUNC_NAME[] = "MPI_Comm_accept";
|
|
|
|
|
|
int MPI_Comm_accept(const char *port_name, MPI_Info info, int root,
|
|
MPI_Comm comm, MPI_Comm *newcomm)
|
|
{
|
|
int rank, rc;
|
|
bool send_first=false; /* we receive first */
|
|
ompi_communicator_t *newcomp=MPI_COMM_NULL;
|
|
|
|
MEMCHECKER(
|
|
memchecker_comm(comm);
|
|
);
|
|
|
|
if ( MPI_PARAM_CHECK ) {
|
|
OMPI_ERR_INIT_FINALIZE(FUNC_NAME);
|
|
|
|
if (ompi_comm_invalid (comm)) {
|
|
return OMPI_ERRHANDLER_INVOKE(MPI_COMM_WORLD, MPI_ERR_COMM,
|
|
FUNC_NAME);
|
|
}
|
|
if ( OMPI_COMM_IS_INTER(comm)) {
|
|
return OMPI_ERRHANDLER_INVOKE(comm, MPI_ERR_COMM,
|
|
FUNC_NAME);
|
|
}
|
|
if ( (0 > root) || (ompi_comm_size(comm) <= root) ) {
|
|
return OMPI_ERRHANDLER_INVOKE(comm, MPI_ERR_ARG,
|
|
FUNC_NAME);
|
|
}
|
|
if ( NULL == newcomm ) {
|
|
return OMPI_ERRHANDLER_INVOKE(comm, 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);
|
|
}
|
|
}
|
|
|
|
rank = ompi_comm_rank ( comm );
|
|
if ( MPI_PARAM_CHECK ) {
|
|
if ( rank == root ) {
|
|
if ( NULL == port_name )
|
|
return OMPI_ERRHANDLER_INVOKE(comm, MPI_ERR_ARG,
|
|
FUNC_NAME);
|
|
}
|
|
}
|
|
|
|
/* parse info object. no prefedined values for this function in MPI-2
|
|
* so lets ignore it for the moment.
|
|
* if ( rank == root && MPI_INFO_NULL != info ) {
|
|
* }
|
|
*/
|
|
OPAL_CR_ENTER_LIBRARY();
|
|
|
|
if ( rank == root ) {
|
|
rc = ompi_dpm_connect_accept (comm, root, port_name, send_first,
|
|
&newcomp);
|
|
}
|
|
else {
|
|
rc = ompi_dpm_connect_accept (comm, root, NULL, send_first,
|
|
&newcomp);
|
|
}
|
|
|
|
*newcomm = newcomp;
|
|
OMPI_ERRHANDLER_RETURN(rc, comm, rc, FUNC_NAME );
|
|
}
|