96 строки
2.6 KiB
C
96 строки
2.6 KiB
C
/* -*- C -*-
|
|
*
|
|
* $HEADER$
|
|
*
|
|
*/
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
|
|
#include "opal/dss/dss.h"
|
|
#include "opal/event/event.h"
|
|
|
|
#include "orte/util/proc_info.h"
|
|
#include "orte/util/name_fns.h"
|
|
#include "orte/runtime/orte_globals.h"
|
|
#include "orte/runtime/runtime.h"
|
|
#include "orte/runtime/orte_wait.h"
|
|
#include "orte/mca/errmgr/errmgr.h"
|
|
#include "orte/mca/rmcast/rmcast.h"
|
|
#include "orte/mca/grpcomm/grpcomm.h"
|
|
|
|
static void cbfunc(int channel, opal_buffer_t *buf, void *cbdata);
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
int rc, i;
|
|
char hostname[512];
|
|
pid_t pid;
|
|
opal_buffer_t buf;
|
|
int32_t i32=1;
|
|
|
|
if (0 > (rc = orte_init(ORTE_PROC_NON_MPI))) {
|
|
fprintf(stderr, "orte_nodename: couldn't init orte - error code %d\n", rc);
|
|
return rc;
|
|
}
|
|
|
|
gethostname(hostname, 512);
|
|
pid = getpid();
|
|
|
|
printf("orte_nodename: Node %s Name %s Pid %ld\n",
|
|
hostname, ORTE_NAME_PRINT(ORTE_PROC_MY_NAME), (long)pid);
|
|
|
|
|
|
OBJ_CONSTRUCT(&buf, opal_buffer_t);
|
|
|
|
if (0 == ORTE_PROC_MY_NAME->vpid) {
|
|
orte_grpcomm.barrier();
|
|
fprintf(stderr, "%d: past barrier\n", (int)ORTE_PROC_MY_NAME->vpid);
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_rmcast.recv_nb(0, ORTE_RMCAST_PERSISTENT,
|
|
ORTE_RMCAST_TAG_WILDCARD,
|
|
cbfunc, NULL))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
}
|
|
opal_dss.pack(&buf, &i32, 1, OPAL_INT32);
|
|
if (ORTE_SUCCESS != (rc = orte_rmcast.send(0, ORTE_RMCAST_TAG_WILDCARD, &buf))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
goto blast;
|
|
}
|
|
} else {
|
|
if (ORTE_SUCCESS != (rc = orte_rmcast.recv_nb(0, ORTE_RMCAST_PERSISTENT,
|
|
ORTE_RMCAST_TAG_WILDCARD,
|
|
cbfunc, NULL))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
}
|
|
orte_grpcomm.barrier();
|
|
fprintf(stderr, "%d: past barrier\n", (int)ORTE_PROC_MY_NAME->vpid);
|
|
|
|
}
|
|
opal_event_dispatch();
|
|
|
|
blast:
|
|
OBJ_DESTRUCT(&buf);
|
|
|
|
orte_finalize();
|
|
return 0;
|
|
}
|
|
|
|
static void cbfunc(int channel, opal_buffer_t *buffer, void *cbdata)
|
|
{
|
|
opal_buffer_t buf;
|
|
int32_t i32=1, rc;
|
|
|
|
fprintf(stderr, "GOT MESSAGE\n");
|
|
fflush(stderr);
|
|
if (0 != ORTE_PROC_MY_NAME->vpid) {
|
|
OBJ_CONSTRUCT(&buf, opal_buffer_t);
|
|
opal_dss.pack(&buf, &i32, 1, OPAL_INT32);
|
|
if (ORTE_SUCCESS != (rc = orte_rmcast.send(0, ORTE_RMCAST_TAG_WILDCARD, &buf))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
}
|
|
OBJ_DESTRUCT(&buf);
|
|
}
|
|
orte_finalize();
|
|
exit(0);
|
|
}
|