2004-01-21 03:05:46 +03:00
|
|
|
/*
|
2005-11-05 22:57:48 +03:00
|
|
|
* Copyright (c) 2004-2005 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.
|
2004-11-28 23:09:25 +03:00
|
|
|
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
|
|
|
|
* University of Stuttgart. All rights reserved.
|
2005-03-24 15:43:37 +03:00
|
|
|
* Copyright (c) 2004-2005 The Regents of the University of California.
|
|
|
|
* All rights reserved.
|
2006-05-18 22:05:46 +04:00
|
|
|
* Copyright (c) 2006 Cisco Systems, Inc. All rights reserved.
|
2004-11-22 04:38:40 +03:00
|
|
|
* $COPYRIGHT$
|
|
|
|
*
|
|
|
|
* Additional copyrights may follow
|
|
|
|
*
|
2004-10-15 23:31:47 +04:00
|
|
|
* $HEADER$
|
2004-01-21 03:05:46 +03:00
|
|
|
*/
|
2004-06-07 19:33:53 +04:00
|
|
|
#include "ompi_config.h"
|
2004-01-21 03:05:46 +03:00
|
|
|
#include <stdio.h>
|
|
|
|
|
2006-02-12 04:33:29 +03:00
|
|
|
#include "ompi/mpi/c/bindings.h"
|
2005-09-12 13:17:44 +04:00
|
|
|
#include "orte/mca/oob/oob.h"
|
|
|
|
#include "ompi/info/info.h"
|
2004-01-21 03:05:46 +03:00
|
|
|
|
2004-06-07 19:33:53 +04:00
|
|
|
#if OMPI_HAVE_WEAK_SYMBOLS && OMPI_PROFILING_DEFINES
|
2004-01-21 03:05:46 +03:00
|
|
|
#pragma weak MPI_Comm_spawn = PMPI_Comm_spawn
|
|
|
|
#endif
|
|
|
|
|
2004-06-07 19:33:53 +04:00
|
|
|
#if OMPI_PROFILING_DEFINES
|
2006-02-12 04:33:29 +03:00
|
|
|
#include "ompi/mpi/c/profile/defines.h"
|
2004-04-20 22:50:43 +04:00
|
|
|
#endif
|
|
|
|
|
2004-07-30 06:58:53 +04:00
|
|
|
static const char FUNC_NAME[] = "MPI_Comm_spawn";
|
2004-06-23 00:21:35 +04:00
|
|
|
|
|
|
|
|
2004-01-21 03:05:46 +03:00
|
|
|
int MPI_Comm_spawn(char *command, char **argv, int maxprocs, MPI_Info info,
|
|
|
|
int root, MPI_Comm comm, MPI_Comm *intercomm,
|
2004-06-16 05:41:01 +04:00
|
|
|
int *array_of_errcodes)
|
|
|
|
{
|
2005-03-14 23:57:21 +03:00
|
|
|
int rank, rc, i;
|
2004-08-05 20:32:13 +04:00
|
|
|
int send_first=0; /* we wait to be contacted */
|
|
|
|
ompi_communicator_t *newcomp;
|
2004-09-29 16:41:55 +04:00
|
|
|
char port_name[MPI_MAX_PORT_NAME];
|
2004-10-01 18:06:23 +04:00
|
|
|
char *tmp_port;
|
2005-03-14 23:57:21 +03:00
|
|
|
orte_rml_tag_t tag;
|
2004-06-17 02:44:56 +04:00
|
|
|
|
2004-06-16 05:41:01 +04:00
|
|
|
if ( MPI_PARAM_CHECK ) {
|
2004-06-23 00:21:35 +04:00
|
|
|
OMPI_ERR_INIT_FINALIZE(FUNC_NAME);
|
2004-06-18 23:02:52 +04:00
|
|
|
|
2006-05-18 22:05:46 +04:00
|
|
|
if ( ompi_comm_invalid (comm)) {
|
2004-06-16 05:41:01 +04:00
|
|
|
return OMPI_ERRHANDLER_INVOKE(MPI_COMM_WORLD, MPI_ERR_COMM,
|
2004-06-23 00:21:35 +04:00
|
|
|
FUNC_NAME);
|
2004-07-30 06:58:53 +04:00
|
|
|
}
|
|
|
|
if ( OMPI_COMM_IS_INTER(comm)) {
|
2004-06-17 01:35:31 +04:00
|
|
|
return OMPI_ERRHANDLER_INVOKE(comm, MPI_ERR_COMM,
|
2004-06-23 00:21:35 +04:00
|
|
|
FUNC_NAME);
|
2004-07-30 06:58:53 +04:00
|
|
|
}
|
2005-01-29 06:01:28 +03:00
|
|
|
if ( (0 > root) || (ompi_comm_size(comm) <= root) ) {
|
2004-06-16 05:41:01 +04:00
|
|
|
return OMPI_ERRHANDLER_INVOKE(comm, MPI_ERR_ARG,
|
2004-06-23 00:21:35 +04:00
|
|
|
FUNC_NAME);
|
2004-07-30 06:58:53 +04:00
|
|
|
}
|
|
|
|
if ( NULL == intercomm ) {
|
2004-06-16 05:41:01 +04:00
|
|
|
return OMPI_ERRHANDLER_INVOKE(comm, MPI_ERR_ARG,
|
2004-06-23 00:21:35 +04:00
|
|
|
FUNC_NAME);
|
2004-07-30 06:58:53 +04:00
|
|
|
}
|
2004-08-12 20:56:24 +04:00
|
|
|
if (NULL == info || ompi_info_is_freed(info)) {
|
|
|
|
return OMPI_ERRHANDLER_INVOKE(MPI_COMM_WORLD, MPI_ERR_INFO,
|
|
|
|
FUNC_NAME);
|
|
|
|
}
|
2004-06-16 05:41:01 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
rank = ompi_comm_rank ( comm );
|
|
|
|
if ( MPI_PARAM_CHECK ) {
|
|
|
|
if ( rank == root ) {
|
2004-07-30 06:58:53 +04:00
|
|
|
if ( NULL == command ) {
|
2004-06-16 05:41:01 +04:00
|
|
|
return OMPI_ERRHANDLER_INVOKE(comm, MPI_ERR_ARG,
|
2004-06-23 00:21:35 +04:00
|
|
|
FUNC_NAME);
|
2004-07-30 06:58:53 +04:00
|
|
|
}
|
|
|
|
if ( 0 > maxprocs ) {
|
2004-06-16 05:41:01 +04:00
|
|
|
return OMPI_ERRHANDLER_INVOKE(comm, MPI_ERR_ARG,
|
2004-06-23 00:21:35 +04:00
|
|
|
FUNC_NAME);
|
2004-07-30 06:58:53 +04:00
|
|
|
}
|
2004-06-16 05:41:01 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2004-06-17 01:35:31 +04:00
|
|
|
|
2004-09-29 16:41:55 +04:00
|
|
|
if ( rank == root ) {
|
|
|
|
/* Open a port. The port_name is passed as an environment variable
|
|
|
|
to the children. */
|
|
|
|
ompi_open_port (port_name);
|
2004-12-16 18:42:02 +03:00
|
|
|
ompi_comm_start_processes (1, &command, &argv, &maxprocs,
|
|
|
|
&info, port_name);
|
2005-01-18 16:56:21 +03:00
|
|
|
tmp_port = ompi_parse_port (port_name, &tag);
|
2004-10-01 18:06:23 +04:00
|
|
|
free(tmp_port);
|
2004-06-17 01:35:31 +04:00
|
|
|
}
|
2004-09-29 16:41:55 +04:00
|
|
|
|
|
|
|
|
2004-09-17 14:12:31 +04:00
|
|
|
rc = ompi_comm_connect_accept (comm, root, NULL, send_first, &newcomp, tag);
|
2004-06-17 01:35:31 +04:00
|
|
|
|
2004-09-29 16:41:55 +04:00
|
|
|
/* close the port again. Nothing has to be done for that at the moment.*/
|
2004-06-17 01:35:31 +04:00
|
|
|
|
|
|
|
/* set error codes */
|
|
|
|
if (MPI_ERRCODES_IGNORE != array_of_errcodes) {
|
|
|
|
for ( i=0; i < maxprocs; i++ ) {
|
2004-08-05 20:32:13 +04:00
|
|
|
array_of_errcodes[i]=rc;
|
2004-06-17 01:35:31 +04:00
|
|
|
}
|
|
|
|
}
|
2004-06-16 05:41:01 +04:00
|
|
|
|
2004-06-17 01:35:31 +04:00
|
|
|
*intercomm = newcomp;
|
2004-08-05 20:32:13 +04:00
|
|
|
OMPI_ERRHANDLER_RETURN (rc, comm, rc, FUNC_NAME);
|
2004-01-21 03:05:46 +03:00
|
|
|
}
|