6e5d844c36
1. completely and cleanly separates responsibilities between the HNP, orted, and tool components. 2. removes all wireup messaging during launch and shutdown. 3. maintains flow control for stdin to avoid large-scale consumption of memory by orteds when large input files are forwarded. This is done using an xon/xoff protocol. 4. enables specification of stdin recipients on the mpirun cmd line. Allowed options include rank, "all", or "none". Default is rank 0. 5. creates a new MPI_Info key "ompi_stdin_target" that supports the above options for child jobs. Default is "none". 6. adds a new tool "orte-iof" that can connect to a running mpirun and display the output. Cmd line options allow selection of any combination of stdout, stderr, and stddiag. Default is stdout. 7. adds a new mpirun and orte-iof cmd line option "tag-output" that will tag each line of output with process name and stream ident. For example, "[1,0]<stdout>this is output" This is not intended for the 1.3 release as it is a major change requiring considerable soak time. This commit was SVN r19767.
59 строки
1.8 KiB
C
59 строки
1.8 KiB
C
#include <stdio.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include <mpi.h>
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
int msg, rc;
|
|
MPI_Comm parent, child;
|
|
int rank, size;
|
|
char hostname[512];
|
|
pid_t pid;
|
|
|
|
pid = getpid();
|
|
printf("Parent [pid %ld] starting up!\n", (long)pid);
|
|
MPI_Init(NULL, NULL);
|
|
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
|
|
printf("%d completed MPI_Init\n", rank);
|
|
MPI_Comm_size(MPI_COMM_WORLD, &size);
|
|
MPI_Comm_get_parent(&parent);
|
|
/* If we get COMM_NULL back, then we're the parent */
|
|
if (MPI_COMM_NULL == parent) {
|
|
pid = getpid();
|
|
printf("Parent [pid %ld] about to spawn!\n", (long)pid);
|
|
if (MPI_SUCCESS != (rc = MPI_Comm_spawn(argv[0], MPI_ARGV_NULL, 3, MPI_INFO_NULL,
|
|
0, MPI_COMM_WORLD, &child, MPI_ERRCODES_IGNORE))) {
|
|
printf("Child failed to spawn\n");
|
|
return rc;
|
|
}
|
|
printf("Parent done with spawn\n");
|
|
if (0 == rank) {
|
|
msg = 38;
|
|
printf("Parent sending message to child\n");
|
|
MPI_Send(&msg, 1, MPI_INT, 0, 1, child);
|
|
}
|
|
MPI_Comm_disconnect(&child);
|
|
printf("Parent disconnected\n");
|
|
}
|
|
/* Otherwise, we're the child */
|
|
else {
|
|
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
|
|
MPI_Comm_size(MPI_COMM_WORLD, &size);
|
|
gethostname(hostname, 512);
|
|
pid = getpid();
|
|
printf("Hello from the child %d of %d on host %s pid %ld\n", rank, 3, hostname, (long)pid);
|
|
if (0 == rank) {
|
|
MPI_Recv(&msg, 1, MPI_INT, 0, 1, parent, MPI_STATUS_IGNORE);
|
|
printf("Child %d received msg: %d\n", rank, msg);
|
|
}
|
|
MPI_Comm_disconnect(&parent);
|
|
printf("Child %d disconnected\n", rank);
|
|
}
|
|
|
|
MPI_Finalize();
|
|
fprintf(stderr, "%d: exiting\n", pid);
|
|
return 0;
|
|
}
|