From 38346e5bdd319bf8d519defce6e368dcc2b89410 Mon Sep 17 00:00:00 2001 From: Edgar Gabriel Date: Tue, 29 Apr 2025 16:22:46 -0500 Subject: [PATCH] use correct MPI initialization method (#90) * use correct MPI initialization method rocSHMEM requires that the MPI library is initialized using THREAD_MULTIPLE support. Lets use that function therefore in our examples. * Update examples/rocshmem_init_attr_test.cc Co-authored-by: Aurelien Bouteiller --------- Co-authored-by: Aurelien Bouteiller [ROCm/rocshmem commit: 2e01af22ca78faf2ebe7a1ef49081c2e048e472d] --- projects/rocshmem/examples/rocshmem_init_attr_test.cc | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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);