From 2f0739d8234eafe8852309cc806dfd8f802bd381 Mon Sep 17 00:00:00 2001
From: avinashkethineedi <avinash.kethineedi@amd.com>
Date: Tue, 1 Oct 2024 20:05:15 +0000
Subject: [PATCH] Add MPI_THREAD_MULTIPLE check

---
 tests/unit_tests/shmem_gtest.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/unit_tests/shmem_gtest.cpp b/tests/unit_tests/shmem_gtest.cpp
index d5a982f4c7..8536ca638a 100644
--- a/tests/unit_tests/shmem_gtest.cpp
+++ b/tests/unit_tests/shmem_gtest.cpp
@@ -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();