Update the orte_mcast test
This commit was SVN r22214.
Этот коммит содержится в:
Ralph Castain
родитель
6496ce7212
Коммит
a2f3a47b92
@ -1,4 +1,4 @@
|
||||
PROGS = no_op sigusr_trap spin orte_nodename orte_spawn orte_loop_spawn orte_loop_child orte_abort get_limits orte_ring spawn_child orte_tool orte_no_op binom oob_stress iof_stress iof_delay radix orte_barrier orte_mcast opal_interface
|
||||
PROGS = no_op sigusr_trap spin orte_nodename orte_spawn orte_loop_spawn orte_loop_child orte_abort get_limits orte_ring spawn_child orte_tool orte_no_op binom oob_stress iof_stress iof_delay radix orte_barrier orte_mcast opal_interface mcast mcast_recv
|
||||
|
||||
all: $(PROGS)
|
||||
|
||||
|
||||
@ -18,15 +18,21 @@
|
||||
#include "orte/mca/rmcast/rmcast.h"
|
||||
#include "orte/mca/grpcomm/grpcomm.h"
|
||||
|
||||
static void cbfunc(int channel, opal_buffer_t *buf, void *cbdata);
|
||||
static void cbfunc(orte_rmcast_channel_t channel, orte_process_name_t *sender, opal_buffer_t *buf, void *cbdata);
|
||||
static void cbfunc_buf_snt(orte_rmcast_channel_t channel, orte_process_name_t *sender, opal_buffer_t *buf, void *cbdata);
|
||||
|
||||
static void cbfunc_iovec(orte_rmcast_channel_t channel,
|
||||
orte_process_name_t *sender,
|
||||
struct iovec *msg, int count, void* cbdata);
|
||||
|
||||
int main(int argc, char* argv[])
|
||||
{
|
||||
int rc, i;
|
||||
char hostname[512];
|
||||
pid_t pid;
|
||||
opal_buffer_t buf;
|
||||
opal_buffer_t buf, *bfptr;
|
||||
int32_t i32=1;
|
||||
struct iovec iovec_array[3];
|
||||
|
||||
if (0 > (rc = orte_init(ORTE_PROC_NON_MPI))) {
|
||||
fprintf(stderr, "orte_nodename: couldn't init orte - error code %d\n", rc);
|
||||
@ -36,60 +42,141 @@ int main(int argc, char* argv[])
|
||||
gethostname(hostname, 512);
|
||||
pid = getpid();
|
||||
|
||||
printf("orte_nodename: Node %s Name %s Pid %ld\n",
|
||||
printf("orte_mcast: 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);
|
||||
}
|
||||
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))) {
|
||||
if (ORTE_SUCCESS != (rc = orte_rmcast.send_buffer(ORTE_RMCAST_APP_PUBLIC_CHANNEL,
|
||||
ORTE_RMCAST_TAG_WILDCARD, &buf))) {
|
||||
ORTE_ERROR_LOG(rc);
|
||||
OBJ_DESTRUCT(&buf);
|
||||
goto blast;
|
||||
}
|
||||
orte_grpcomm.barrier();
|
||||
OBJ_DESTRUCT(&buf);
|
||||
bfptr = OBJ_NEW(opal_buffer_t);
|
||||
i32 = 2;
|
||||
opal_dss.pack(bfptr, &i32, 1, OPAL_INT32);
|
||||
if (ORTE_SUCCESS != (rc = orte_rmcast.send_buffer_nb(ORTE_RMCAST_APP_PUBLIC_CHANNEL,
|
||||
ORTE_RMCAST_TAG_WILDCARD, bfptr,
|
||||
cbfunc_buf_snt, NULL))) {
|
||||
ORTE_ERROR_LOG(rc);
|
||||
OBJ_RELEASE(bfptr);
|
||||
goto blast;
|
||||
}
|
||||
orte_grpcomm.barrier();
|
||||
/* create an iovec array */
|
||||
for (i=0; i < 3; i++) {
|
||||
iovec_array[i].iov_base = (uint8_t*)malloc(30);
|
||||
iovec_array[i].iov_len = 30;
|
||||
}
|
||||
/* send it out */
|
||||
if (ORTE_SUCCESS != (rc = orte_rmcast.send(ORTE_RMCAST_APP_PUBLIC_CHANNEL,
|
||||
ORTE_RMCAST_TAG_WILDCARD,
|
||||
iovec_array, 3))) {
|
||||
ORTE_ERROR_LOG(rc);
|
||||
goto blast;
|
||||
}
|
||||
orte_grpcomm.barrier();
|
||||
orte_finalize();
|
||||
return 0;
|
||||
} else {
|
||||
if (ORTE_SUCCESS != (rc = orte_rmcast.recv_nb(0, ORTE_RMCAST_PERSISTENT,
|
||||
ORTE_RMCAST_TAG_WILDCARD,
|
||||
cbfunc, NULL))) {
|
||||
if (ORTE_SUCCESS != (rc = orte_rmcast.recv_buffer_nb(ORTE_RMCAST_APP_PUBLIC_CHANNEL,
|
||||
ORTE_RMCAST_PERSISTENT,
|
||||
ORTE_RMCAST_TAG_WILDCARD,
|
||||
cbfunc, NULL))) {
|
||||
ORTE_ERROR_LOG(rc);
|
||||
}
|
||||
if (ORTE_SUCCESS != (rc = orte_rmcast.recv_nb(ORTE_RMCAST_APP_PUBLIC_CHANNEL,
|
||||
ORTE_RMCAST_PERSISTENT,
|
||||
ORTE_RMCAST_TAG_WILDCARD,
|
||||
cbfunc_iovec, 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);
|
||||
|
||||
blast:
|
||||
orte_finalize();
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void cbfunc(int channel, opal_buffer_t *buffer, void *cbdata)
|
||||
static void cbfunc(orte_rmcast_channel_t channel, orte_process_name_t *sender, opal_buffer_t *buffer, void *cbdata)
|
||||
{
|
||||
opal_buffer_t buf;
|
||||
int32_t i32=1, rc;
|
||||
int32_t i32, rc;
|
||||
|
||||
fprintf(stderr, "GOT MESSAGE\n");
|
||||
/* retrieve the value sent */
|
||||
rc = 1;
|
||||
opal_dss.unpack(buffer, &i32, &rc, OPAL_INT32);
|
||||
|
||||
fprintf(stderr, "%s GOT BUFFER MESSAGE from %s with value %d\n",
|
||||
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
|
||||
ORTE_NAME_PRINT(sender), i32);
|
||||
fflush(stderr);
|
||||
|
||||
orte_grpcomm.barrier();
|
||||
|
||||
#if 0
|
||||
opal_buffer_t *buf;
|
||||
int32_t i32=2, rc;
|
||||
|
||||
buf = OBJ_NEW(opal_buffer_t);
|
||||
opal_dss.pack(buf, &i32, 1, OPAL_INT32);
|
||||
opal_dss.pack(buf, &i32, 1, OPAL_INT32);
|
||||
opal_dss.pack(buf, &i32, 1, OPAL_INT32);
|
||||
|
||||
|
||||
|
||||
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);
|
||||
if (1 == i32) {
|
||||
if (ORTE_SUCCESS != (rc = orte_rmcast.send_buffer(ORTE_RMCAST_APP_PUBLIC_CHANNEL,
|
||||
ORTE_RMCAST_TAG_WILDCARD, buf))) {
|
||||
ORTE_ERROR_LOG(rc);
|
||||
}
|
||||
}
|
||||
OBJ_DESTRUCT(&buf);
|
||||
OBJ_RELEASE(buf);
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
static void cbfunc_iovec(orte_rmcast_channel_t channel,
|
||||
orte_process_name_t *sender,
|
||||
struct iovec *msg, int count, void* cbdata)
|
||||
{
|
||||
int rc;
|
||||
|
||||
fprintf(stderr, "%s GOT IOVEC MESSAGE from %s of %d elements\n",
|
||||
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME), ORTE_NAME_PRINT(sender), count);
|
||||
fflush(stderr);
|
||||
#if 0
|
||||
if (0 != ORTE_PROC_MY_NAME->vpid) {
|
||||
/* send it back */
|
||||
if (ORTE_SUCCESS != (rc = orte_rmcast.send(ORTE_RMCAST_APP_PUBLIC_CHANNEL,
|
||||
ORTE_RMCAST_TAG_WILDCARD, msg, count))) {
|
||||
ORTE_ERROR_LOG(rc);
|
||||
}
|
||||
}
|
||||
#endif
|
||||
orte_grpcomm.barrier();
|
||||
orte_finalize();
|
||||
exit(0);
|
||||
}
|
||||
|
||||
static void cbfunc_buf_snt(orte_rmcast_channel_t channel, orte_process_name_t *sender, opal_buffer_t *buf, void *cbdata)
|
||||
{
|
||||
fprintf(stderr, "%s BUFFERED_NB SEND COMPLETE\n", ORTE_NAME_PRINT(ORTE_PROC_MY_NAME));
|
||||
fflush(stderr);
|
||||
|
||||
OBJ_RELEASE(buf);
|
||||
}
|
||||
|
||||
Загрузка…
x
Ссылка в новой задаче
Block a user