/*---------------------------------------------------------------------------*/
na_bool_t
NA_Addr_is_self(na_class_t *na_class, na_addr_t addr)
{
assert(na_class);
if (!na_class->addr_is_self) {
NA_LOG_WARNING("NA_Addr_is_self not supported by plugin");
return NA_FALSE;
}
return na_class->addr_is_self(na_class, addr);
}
static void
na_test_mpi_init(struct na_test_info *na_test_info)
{
int mpi_initialized = 0;
int mpi_finalized = 0;
na_test_info->mpi_comm = MPI_COMM_WORLD; /* default */
MPI_Initialized(&mpi_initialized);
if (mpi_initialized) {
NA_LOG_WARNING("MPI was already initialized");
goto done;
}
MPI_Finalized(&mpi_finalized);
if (mpi_finalized) {
NA_LOG_ERROR("MPI was already finalized");
goto done;
}
#ifdef NA_MPI_HAS_GNI_SETUP
/* Setup GNI job before initializing MPI */
if (NA_MPI_Gni_job_setup() != NA_SUCCESS) {
NA_LOG_ERROR("Could not setup GNI job");
return;
}
#endif
if (na_test_info->listen || na_test_info->mpi_static) {
int provided;
MPI_Init_thread(NULL, NULL, MPI_THREAD_MULTIPLE, &provided);
if (provided != MPI_THREAD_MULTIPLE) {
NA_LOG_ERROR("MPI_THREAD_MULTIPLE cannot be set");
}
/* Only if we do static MPMD MPI */
if (na_test_info->mpi_static) {
int mpi_ret, color, global_rank;
MPI_Comm_rank(MPI_COMM_WORLD, &global_rank);
/* Color is 1 for server, 2 for client */
color = (na_test_info->listen) ? 1 : 2;
/* Assume that the application did not split MPI_COMM_WORLD already */
mpi_ret = MPI_Comm_split(MPI_COMM_WORLD, color, global_rank,
&na_test_info->mpi_comm);
if (mpi_ret != MPI_SUCCESS) {
NA_LOG_ERROR("Could not split communicator");
}
#ifdef NA_HAS_MPI
/* Set init comm that will be used to setup NA MPI */
NA_MPI_Set_init_intra_comm(na_test_info->mpi_comm);
#endif
}
} else {
MPI_Init(NULL, NULL);
}
done:
MPI_Comm_rank(na_test_info->mpi_comm, &na_test_info->mpi_comm_rank);
MPI_Comm_size(na_test_info->mpi_comm, &na_test_info->mpi_comm_size);
return;
}
