explicit start of hpctoolkit sampling

mochi-hpc · Dec 16, 2024 · bca857f · bca857f
1 parent 314c4b9
commit bca857f
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/quintain-server.c b/src/quintain-server.c
@@ -18,6 +18,10 @@
 #include <margo-bulk-pool.h>
 #include <quintain-server.h>
 
+#ifdef HAVE_HPCTOOLKIT
+    #include <hpctoolkit.h>
+#endif
+
 #include "quintain-rpc.h"
 #include "quintain-macros.h"
 
@@ -172,6 +176,10 @@ int quintain_provider_deregister(quintain_provider_t provider)
     return QTN_SUCCESS;
 }
 
+#ifdef HAVE_HPCTOOLKIT
+static int hpctoolkit_started = 0;
+#endif
+
 static void qtn_work_ult(hg_handle_t handle)
 {
     margo_instance_id     mid = MARGO_INSTANCE_NULL;
@@ -186,6 +194,13 @@ static void qtn_work_ult(hg_handle_t handle)
 
     memset(&out, 0, sizeof(out));
 
+#ifdef HAVE_HPCTOOLKIT
+    if (!hpctoolkit_started) {
+        hpctoolkit_sampling_start();
+        hpctoolkit_started = 1;
+    }
+#endif
+
     mid = margo_hg_handle_get_instance(handle);
     assert(mid);
     info     = margo_get_info(handle);