Adding two more static task wrappers to check for shutdown.

UO-OACISS · Jun 29, 2018 · 45ccbbd · 45ccbbd
1 parent d555ebc
commit 45ccbbd
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/apex/task_wrapper.hpp b/src/apex/task_wrapper.hpp
@@ -49,6 +49,26 @@ struct task_wrapper {
         tt_ptr->task_id = task_identifier::get_task_id(apex_main_str);
         return tt_ptr;
     }
+    static inline std::shared_ptr<task_wrapper> & get_apex_hpx_shutdown_wrapper(void) {
+        static std::shared_ptr<task_wrapper> tt_ptr(nullptr);
+        if (tt_ptr.get() != nullptr) {
+            return tt_ptr;
+        }
+        const std::string apex_shutdown_str("shutdown_all_action");
+        tt_ptr = std::make_shared<task_wrapper>();
+        tt_ptr->task_id = task_identifier::get_task_id(apex_shutdown_str);
+        return tt_ptr;
+    }
+    static inline std::shared_ptr<task_wrapper> & get_apex_hpx_timer_wrapper(void) {
+        static std::shared_ptr<task_wrapper> tt_ptr(nullptr);
+        if (tt_ptr.get() != nullptr) {
+            return tt_ptr;
+        }
+        const std::string apex_timer_str("at_timer (expire at)");
+        tt_ptr = std::make_shared<task_wrapper>();
+        tt_ptr->task_id = task_identifier::get_task_id(apex_timer_str);
+        return tt_ptr;
+    }
 }; // struct task_wrapper
 
 }; // namespace apex