fix for #414 making numa node count atomic

1 files changed, 4 insertions, 3 deletions

diff --git a/include/mimalloc-internal.h b/include/mimalloc-internal.h
index 4a803ff..1e1a796 100644
--- a/include/mimalloc-internal.h
+++ b/include/mimalloc-internal.h
@@ -667,13 +667,14 @@ static inline uintptr_t _mi_random_shuffle(uintptr_t x) {
 int    _mi_os_numa_node_get(mi_os_tld_t* tld);
 size_t _mi_os_numa_node_count_get(void);
 
-extern size_t _mi_numa_node_count;
+extern _Atomic(size_t) _mi_numa_node_count;
 static inline int _mi_os_numa_node(mi_os_tld_t* tld) {
-  if (mi_likely(_mi_numa_node_count == 1)) return 0;
+  if (mi_likely(mi_atomic_load_relaxed(&_mi_numa_node_count) == 1)) return 0;
   else return _mi_os_numa_node_get(tld);
 }
 static inline size_t _mi_os_numa_node_count(void) {
-  if (mi_likely(_mi_numa_node_count>0)) return _mi_numa_node_count;
+  const size_t count = mi_atomic_load_relaxed(&_mi_numa_node_count);
+  if (mi_likely(count>0)) return count;
   else return _mi_os_numa_node_count_get();
 }