From bcdb4dc639c94d4d80968de94ace435183875ae1 Mon Sep 17 00:00:00 2001
From: Ruben Sturm <ruben>
Date: Mon, 2 Dec 2024 18:10:21 +0000
Subject: [PATCH] mimalloc fix

---
 src/init.c | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/init.c b/src/init.c
index 6f51ca89..58a0428b 100644
--- a/src/init.c
+++ b/src/init.c
@@ -462,14 +462,15 @@ void _mi_heap_set_default_direct(mi_heap_t* heap)  {
   #elif defined(MI_TLS_PTHREAD_SLOT_OFS)
   *mi_prim_tls_pthread_heap_slot() = heap;
   #elif defined(MI_TLS_PTHREAD)
+    // ensure the default heap is passed to `_mi_thread_done`
+  // setting to a non-NULL value also ensures `mi_thread_done` is called.
+  _mi_prim_thread_associate_default_heap(heap);
   // we use _mi_heap_default_key
   #else
   _mi_heap_default = heap;
   #endif
 
-  // ensure the default heap is passed to `_mi_thread_done`
-  // setting to a non-NULL value also ensures `mi_thread_done` is called.
-  _mi_prim_thread_associate_default_heap(heap);
+
 }