Add MPI_THREAD_MULTIPLE check

[ROCm/rocshmem commit: 2f0739d823]
2024-10-01 20:05:15 +00:00
parent 0641a4a29e
commit 285ac5cab6
1 changed files with 3 additions and 0 deletions
@@ -31,6 +31,9 @@ int main(int argc, char **argv) {
  if (!initialized) {
    int provided;
    MPI_Init_thread(nullptr, nullptr, MPI_THREAD_MULTIPLE, &provided);
+    if (provided != MPI_THREAD_MULTIPLE) {
+      std::cerr << "MPI_THREAD_MULTIPLE support disabled.\n";
+    }
  }

  int ret_val = RUN_ALL_TESTS();