1
1
openmpi/orte/test/mpi/hello_nodename.c
Karol Mroz 5c11bdb251 orte: fixup hostname max length usage
Also removes orte specific max hostname value.

Signed-off-by: Karol Mroz <mroz.karol@gmail.com>
2016-04-25 07:08:23 +02:00

53 строки
1.2 KiB
C

/* -*- C -*-
*
* $HEADER$
*
* The most basic of MPI applications
*/
#define _GNU_SOURCE
#include "orte_config.h"
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include "mpi.h"
int main(int argc, char* argv[])
{
int rank, size;
char hostname[OPAL_MAXHOSTNAMELEN];
void *appnum;
void *univ_size;
char *appstr, *unistr;
int flag;
char *envar;
envar = getenv("OMPI_UNIVERSE_SIZE");
MPI_Init(&argc, &argv);
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
MPI_Comm_size(MPI_COMM_WORLD, &size);
MPI_Comm_get_attr(MPI_COMM_WORLD, MPI_APPNUM, &appnum, &flag);
if (NULL == appnum) {
asprintf(&appstr, "UNDEFINED");
} else {
asprintf(&appstr, "%d", *(int*)appnum);
}
MPI_Comm_get_attr(MPI_COMM_WORLD, MPI_UNIVERSE_SIZE, &univ_size, &flag);
if (NULL == univ_size) {
asprintf(&unistr, "UNDEFINED");
} else {
asprintf(&unistr, "%d", *(int*)univ_size);
}
gethostname(hostname, sizeof(hostname));
printf("Hello, World, I am %d of %d on host %s from app number %s universe size %s universe envar %s\n",
rank, size, hostname, appstr, unistr, (NULL == envar) ? "NULL" : envar);
MPI_Finalize();
return 0;
}