Merge pull request #704 from msimberg/pika-wait

Add `pika::wait`
pika-org · Jul 12, 2023 · 419e357 · 419e357
2 parents 03156cd + c7fcdf5
commit 419e357
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/libs/pika/init_runtime/include/pika/init_runtime/init_runtime.hpp b/libs/pika/init_runtime/include/pika/init_runtime/init_runtime.hpp
@@ -143,6 +143,7 @@ namespace pika {
         init_params const& params = init_params());
     PIKA_EXPORT int finalize(error_code& ec = throws);
     PIKA_EXPORT int stop(error_code& ec = throws);
+    PIKA_EXPORT int wait(error_code& ec = throws);
     PIKA_EXPORT int suspend(error_code& ec = throws);
     PIKA_EXPORT int resume(error_code& ec = throws);
 }    // namespace pika
diff --git a/libs/pika/init_runtime/src/init_runtime.cpp b/libs/pika/init_runtime/src/init_runtime.cpp
@@ -217,6 +217,21 @@ namespace pika {
         return result;
     }
 
+    int wait(error_code& ec)
+    {
+        runtime* rt = get_runtime_ptr();
+        if (nullptr == rt)
+        {
+            PIKA_THROWS_IF(ec, pika::error::invalid_status, "pika::wait",
+                "the runtime system is not active (did you already call pika::stop?)");
+            return -1;
+        }
+
+        rt->get_thread_manager().wait();
+
+        return 0;
+    }
+
     int suspend(error_code& ec)
     {
         if (threads::detail::get_self_ptr())