5ef6367f73
Specifically, this patch makes the following changes. build scripts: - master_top_srcdir -> main_top_srcdir Git: - "master" -> "main" branch (variable names and comments) global structures and variables: - MPIR_ThreadInfo.master_thread -> MPIR_ThreadInfo.main_thread - struct PMIMaster -> struct PMIMain - PMI2_Connect_comm_t.isMaster -> PMI2_Connect_comm_t.isMain names of tests: - spawntest_master -> spawntest_parent - taskmaster -> taskmanager - th_taskmaster -> th_taskmanager` comments and names of temporary/local variables: - master -> main, parent, server, ... - slave -> child, worker, client, ...
113 linhas
3.2 KiB
C
113 linhas
3.2 KiB
C
/*
|
|
* Copyright (C) by Argonne National Laboratory
|
|
* See COPYRIGHT in top-level directory
|
|
*/
|
|
|
|
#include "mpi.h"
|
|
#include "mpitest.h"
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#ifdef HAVE_STRINGS_H
|
|
#include <strings.h>
|
|
#endif
|
|
#ifdef HAVE_STRING_H
|
|
#include <string.h>
|
|
#endif
|
|
|
|
#define IF_VERBOSE(a) if (verbose) { printf a ; fflush(stdout); }
|
|
|
|
/*
|
|
static char MTEST_Descrip[] = "A simple test of Comm_disconnect";
|
|
*/
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
int errs = 0;
|
|
int rank, size, rsize;
|
|
int np = 3;
|
|
MPI_Comm parentcomm, intercomm;
|
|
int verbose = 0;
|
|
char *env;
|
|
int can_spawn;
|
|
|
|
env = getenv("MPITEST_VERBOSE");
|
|
if (env) {
|
|
if (*env != '0')
|
|
verbose = 1;
|
|
}
|
|
|
|
MTest_Init(&argc, &argv);
|
|
|
|
errs += MTestSpawnPossible(&can_spawn);
|
|
|
|
if (can_spawn) {
|
|
MPI_Comm_get_parent(&parentcomm);
|
|
|
|
if (parentcomm == MPI_COMM_NULL) {
|
|
IF_VERBOSE(("spawning %d processes\n", np));
|
|
/* Create 3 more processes */
|
|
MPI_Comm_spawn((char *) "./disconnect", MPI_ARGV_NULL, np,
|
|
MPI_INFO_NULL, 0, MPI_COMM_WORLD, &intercomm, MPI_ERRCODES_IGNORE);
|
|
} else {
|
|
intercomm = parentcomm;
|
|
}
|
|
|
|
/* We now have a valid intercomm */
|
|
|
|
MPI_Comm_remote_size(intercomm, &rsize);
|
|
MPI_Comm_size(intercomm, &size);
|
|
MPI_Comm_rank(intercomm, &rank);
|
|
|
|
if (parentcomm == MPI_COMM_NULL) {
|
|
IF_VERBOSE(("parent rank %d alive.\n", rank));
|
|
/* Parent */
|
|
if (rsize != np) {
|
|
errs++;
|
|
printf("Did not create %d processes (got %d)\n", np, rsize);
|
|
fflush(stdout);
|
|
}
|
|
IF_VERBOSE(("disconnecting child communicator\n"));
|
|
MPI_Comm_disconnect(&intercomm);
|
|
|
|
/* Errors cannot be sent back to the parent because there is no
|
|
* communicator connected to the children
|
|
* for (i=0; i<rsize; i++)
|
|
* {
|
|
* MPI_Recv(&err, 1, MPI_INT, i, 1, intercomm, MPI_STATUS_IGNORE);
|
|
* errs += err;
|
|
* }
|
|
*/
|
|
} else {
|
|
IF_VERBOSE(("child rank %d alive.\n", rank));
|
|
/* Child */
|
|
if (size != np) {
|
|
errs++;
|
|
printf("(Child) Did not create %d processes (got %d)\n", np, size);
|
|
fflush(stdout);
|
|
}
|
|
|
|
IF_VERBOSE(("disconnecting communicator\n"));
|
|
MPI_Comm_disconnect(&intercomm);
|
|
|
|
/* Send the errs back to the parent process */
|
|
/* Errors cannot be sent back to the parent because there is no
|
|
* communicator connected to the parent */
|
|
/*MPI_Ssend(&errs, 1, MPI_INT, 0, 1, intercomm); */
|
|
}
|
|
|
|
/* Note that the MTest_Finalize get errs only over COMM_WORLD */
|
|
/* Note also that both the parent and child will generate "No Errors"
|
|
* if both call MTest_Finalize */
|
|
if (parentcomm == MPI_COMM_NULL) {
|
|
MTest_Finalize(errs);
|
|
} else {
|
|
MPI_Finalize();
|
|
}
|
|
} else {
|
|
MTest_Finalize(errs);
|
|
}
|
|
|
|
IF_VERBOSE(("calling finalize\n"));
|
|
return MTestReturnValue(errs);
|
|
}
|