Shintaro Iwasaki
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, ...
118 linhas
3.6 KiB
C++
118 linhas
3.6 KiB
C++
/*
|
|
* Copyright (C) by Argonne National Laboratory
|
|
* See COPYRIGHT in top-level directory
|
|
*/
|
|
|
|
#include "mpi.h"
|
|
#include "mpitestconf.h"
|
|
#ifdef HAVE_IOSTREAM
|
|
// Not all C++ compilers have iostream instead of iostream.h
|
|
#include <iostream>
|
|
#ifdef HAVE_NAMESPACE_STD
|
|
// Those that do need the std namespace; otherwise, a bare "cout"
|
|
// is likely to fail to compile
|
|
using namespace std;
|
|
#endif
|
|
#else
|
|
#include <iostream.h>
|
|
#endif
|
|
#include "mpitestcxx.h"
|
|
#ifdef HAVE_STRING_H
|
|
#include <string.h>
|
|
#endif
|
|
|
|
static char MTEST_Descrip[] = "A simple test of Comm_spawn, with complex arguments";
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
int errs = 0, err;
|
|
int rank, size, rsize, i;
|
|
int np = 2;
|
|
int errcodes[2];
|
|
MPI::Intercomm parentcomm, intercomm;
|
|
MPI::Status status;
|
|
const char *inargv[] = { "a", "b=c", "d e", "-pf", " Ss", 0 };
|
|
const char *outargv[] = { "a", "b=c", "d e", "-pf", " Ss", 0 };
|
|
|
|
MTest_Init();
|
|
|
|
parentcomm = MPI::Comm::Get_parent();
|
|
|
|
if (parentcomm == MPI::COMM_NULL) {
|
|
/* Create 2 more processes */
|
|
/* ./ is unix specific .
|
|
* The more generic approach would be to specify "spawnargv" as the
|
|
* executable and pass an info with ("path", ".") */
|
|
intercomm = MPI::COMM_WORLD.Spawn("./spawnargvx", inargv, np, MPI::INFO_NULL, 0, errcodes);
|
|
} else
|
|
intercomm = parentcomm;
|
|
|
|
/* We now have a valid intercomm */
|
|
|
|
rsize = intercomm.Get_remote_size();
|
|
size = intercomm.Get_size();
|
|
rank = intercomm.Get_rank();
|
|
|
|
if (parentcomm == MPI::COMM_NULL) {
|
|
/* Parent */
|
|
if (rsize != np) {
|
|
errs++;
|
|
cout << "Did not create " << np << " processes (got " << rsize << ")\n";
|
|
}
|
|
for (i = 0; i < rsize; i++) {
|
|
intercomm.Send(&i, 1, MPI::INT, i, 0);
|
|
}
|
|
/* We could use intercomm reduce to get the errors from the
|
|
* children, but we'll use a simpler loop to make sure that
|
|
* we get valid data */
|
|
for (i = 0; i < rsize; i++) {
|
|
intercomm.Recv(&err, 1, MPI::INT, i, 1);
|
|
errs += err;
|
|
}
|
|
} else {
|
|
/* Child */
|
|
/* FIXME: This assumes that stdout is handled for the children
|
|
* (the error count will still be reported to the parent) */
|
|
if (size != np) {
|
|
errs++;
|
|
cout << "(Child) Did not create " << np << " (got " << size << ")\n";
|
|
}
|
|
intercomm.Recv(&i, 1, MPI::INT, 0, 0, status);
|
|
if (i != rank) {
|
|
errs++;
|
|
cout << "Unexpected rank on child " << rank << " (" << i << "\n";
|
|
}
|
|
/* Check the command line */
|
|
for (i = 1; i < argc; i++) {
|
|
if (!outargv[i - 1]) {
|
|
errs++;
|
|
cout << "Wrong number of arguments (" << argc << ")\n";
|
|
break;
|
|
}
|
|
if (strcmp(argv[i], outargv[i - 1]) != 0) {
|
|
errs++;
|
|
cout << "Found arg " << argv[i] << " but expected " << outargv[i - 1] << "\n";
|
|
}
|
|
}
|
|
if (outargv[i - 1]) {
|
|
/* We had too few args in the spawned command */
|
|
errs++;
|
|
cout << "Too few arguments to spawned command\n";
|
|
}
|
|
/* Send the errs back to the parent process */
|
|
intercomm.Ssend(&errs, 1, MPI::INT, 0, 1);
|
|
}
|
|
|
|
/* It isn't necessary to free the intercomm, but it should not hurt */
|
|
intercomm.Free();
|
|
|
|
/* Note that the MTest_Finalize get errs only over COMM_WORLD */
|
|
if (parentcomm == MPI::COMM_NULL) {
|
|
MTest_Finalize(errs);
|
|
} else {
|
|
MPI_Finalize();
|
|
}
|
|
|
|
return 0;
|
|
}
|