init thread for heap preinit

lslusarczyk · Oct 23, 2023 · bb6b457 · bb6b457
1 parent a645139
commit bb6b457
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/mpp/shmemx_c_func.h4 b/mpp/shmemx_c_func.h4
@@ -38,4 +38,5 @@ SHMEM_FUNCTION_ATTRIBUTES void SHPRE()shmemx_pcntr_get_all(shmem_ctx_t ctx, shme
 /* Separate initializers */
 SHMEM_FUNCTION_ATTRIBUTES void SHPRE()shmemx_heap_create(void *base, size_t size, int device_type, int device_index);
 SHMEM_FUNCTION_ATTRIBUTES void SHPRE()shmemx_heap_preinit();
+SHMEM_FUNCTION_ATTRIBUTES int SHPRE()shmemx_heap_preinit_thread(int requested, int *provided);
 SHMEM_FUNCTION_ATTRIBUTES void SHPRE()shmemx_heap_postinit();
diff --git a/src/init_c.c b/src/init_c.c
@@ -108,6 +108,19 @@ shmem_init_thread(int tl_requested, int *tl_provided)
 }
 
 
+int SHMEM_FUNCTION_ATTRIBUTES
+shmemx_heap_preinit_thread(int tl_requested, int *tl_provided)
+{
+    int ret;
+    if (shmem_internal_initialized) {
+        RAISE_ERROR_STR("attempt to reinitialize library");
+    }
+
+    ret = shmem_internal_heap_preinit(tl_requested, tl_provided);
+    return ret;
+}
+
+
 void SHMEM_FUNCTION_ATTRIBUTES
 shmem_query_thread(int *provided)
 {