40c0b6b12d
only be used if the RTE init functions have been called. Not quite as flexible as the real waitpid() function (no -1 support), but all I need for the SSH / BProc / RMS pcms. This code is not yet turned on by default (need to add the init / finalize calls to ompi_rte_init?? and ompi_rte_finalize() This commit was SVN r2860.
70 строки
1.2 KiB
C
70 строки
1.2 KiB
C
/*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "mpi.h"
|
|
#include "runtime/ompi_rte_wait.h"
|
|
#include "runtime/ompi_progress.h"
|
|
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
int count = 0;
|
|
|
|
static void
|
|
callback(pid_t pid, int status, void *data)
|
|
{
|
|
printf("callback for %d, %d\n", pid, status);
|
|
count--;
|
|
}
|
|
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
pid_t pid, ret;
|
|
int status = -1;
|
|
|
|
MPI_Init(&argc, &argv);
|
|
ompi_rte_wait_init();
|
|
|
|
pid = fork();
|
|
if (pid > 0) {
|
|
count++;
|
|
ompi_rte_wait_cb(pid, callback, NULL);
|
|
} else {
|
|
sleep(10);
|
|
printf("pid %d exiting\n", getpid());
|
|
exit(0);
|
|
}
|
|
|
|
pid = fork();
|
|
if (pid > 0) {
|
|
ret = ompi_rte_waitpid(pid, &status, 0);
|
|
printf("pid %d waitpid, status %d\n", ret, status);
|
|
} else {
|
|
sleep(5);
|
|
printf("pid %d exiting\n", getpid());
|
|
exit(0);
|
|
}
|
|
|
|
pid = fork();
|
|
if (pid > 0) {
|
|
count++;
|
|
ompi_rte_wait_cb(pid, callback, NULL);
|
|
} else {
|
|
printf("pid %d exiting\n", getpid());
|
|
exit(0);
|
|
}
|
|
|
|
while (count > 0) { ompi_progress(); }
|
|
|
|
ompi_rte_wait_finalize();
|
|
MPI_Finalize();
|
|
|
|
|
|
return 0;
|
|
}
|