initialize hwloc topology only before first fork, not always

Author: lplewa

src/libumf.c

@@ -7,6 +7,7 @@
  *
  */
 
+#include <pthread.h>
 #include <stdarg.h>
 #include <stddef.h>
 #include <string.h>
@@ -40,6 +41,11 @@ static umf_ctl_node_t CTL_NODE(umf)[] = {CTL_CHILD(provider), CTL_CHILD(pool),
 
 void initialize_global_ctl(void) { CTL_REGISTER_MODULE(NULL, umf); }
 
+// Benchmarks may fork multiple times and topology init is slow.
+// Init topology before fork (if not already) so children don't repeat it.
+// TODO: This is a hack. Better solution is needed.
+static void atfork_prepare(void) { umfGetTopologyReduced(); }
+
 umf_result_t umfInit(void) {
     utils_init_once(&initMutexOnce, initialize_init_mutex);
 
@@ -74,11 +80,11 @@ umf_result_t umfInit(void) {
     if (TRACKER) {
         LOG_DEBUG("UMF library initialized");
     }
-#if !defined(UMF_NO_HWLOC)
-    // some benchmarks uses multiple forks, and topology initialization is very slow
-    // so if we initialize topology before the first fork, we can get significant performance gain.
-    umfGetTopologyReduced();
+
+#if !defined(_WIN32) && !defined(UMF_NO_HWLOC)
+    pthread_atfork(atfork_prepare, NULL, NULL);
 #endif
+
     return UMF_RESULT_SUCCESS;
 }