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 <Aurelien.bouteiller@gmail.com>

---------

Co-authored-by: Aurelien Bouteiller <Aurelien.bouteiller@gmail.com>

[ROCm/rocshmem commit: 2e01af22ca]
此提交包含在:
Edgar Gabriel
2025-04-29 16:22:46 -05:00
提交者 GitHub
父節點 2e6fed8e79
當前提交 38346e5bdd
+5 -1
查看文件
@@ -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);