diff --git a/projects/rocshmem/examples/rocshmem_init_attr_test.cc b/projects/rocshmem/examples/rocshmem_init_attr_test.cc index dee1092927..d1b4d77f6c 100644 --- a/projects/rocshmem/examples/rocshmem_init_attr_test.cc +++ b/projects/rocshmem/examples/rocshmem_init_attr_test.cc @@ -58,8 +58,12 @@ int main (int argc, char **argv) int ret; rocshmem_uniqueid_t uid; rocshmem_init_attr_t attr; + int provided; - MPI_Init(&argc, &argv); + MPI_Init_thread (&argc, &argv, MPI_THREAD_MULTIPLE, &provided); + if (provided != MPI_THREAD_MULTIPLE) { + std::cerr << "MPI_THREAD_MULTIPLE support disabled.\n"; + } MPI_Comm_rank (MPI_COMM_WORLD, &world_rank); MPI_Comm_size (MPI_COMM_WORLD, &world_nranks);