Add TSAN annotations to pool_scalable

To fix false positives from TSAN.

Author: igchor

src/pool/pool_scalable.c

@@ -1,6 +1,6 @@
 /*
  *
- * Copyright (C) 2023 Intel Corporation
+ * Copyright (C) 2023-2024 Intel Corporation
  *
  * Under the Apache License v2.0 with LLVM Exceptions. See LICENSE.TXT.
  * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
@@ -24,6 +24,7 @@
 #include <tbb/scalable_allocator.h>
 
 #include "utils_common.h"
+#include "utils_sanitizers.h"
 
 typedef void *(*raw_alloc_tbb_type)(intptr_t, size_t *);
 typedef void (*raw_free_tbb_type)(intptr_t, void *, size_t);
@@ -191,6 +192,13 @@ static void *tbb_calloc(void *pool, size_t num, size_t size) {
         return NULL;
     }
 
+    // synchronizes with release in tbb_free:
+    // if another thread modified memory allocated from tbb and called free
+    // and we end up getting this memory from tbb_malloc above we need to
+    // tell asan that the memset below does not conflict with any preceeding
+    // stores
+    utils_annotate_acquire(ptr);
+
     memset(ptr, 0, csize);
     return ptr;
 }
@@ -229,6 +237,9 @@ static umf_result_t tbb_free(void *pool, void *ptr) {
 
     TLS_last_free_error = UMF_RESULT_SUCCESS;
 
+    // synchronizes (happens-before) with acquire in tbb_calloc
+    utils_annotate_release(ptr);
+
     struct tbb_memory_pool *pool_data = (struct tbb_memory_pool *)pool;
     if (g_tbb_ops.pool_free(pool_data->tbb_pool, ptr)) {
         return UMF_RESULT_SUCCESS;

src/utils/utils_sanitizers.h

@@ -0,0 +1,32 @@
+/*
+ *
+ * Copyright (C) 2024 Intel Corporation
+ *
+ * Under the Apache License v2.0 with LLVM Exceptions. See LICENSE.TXT.
+ * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+ *
+ */
+
+#if __SANITIZE_THREAD__
+#include <sanitizer/tsan_interface.h>
+#endif
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+void utils_annotate_acquire(void *ptr) {
+#if __SANITIZE_THREAD__
+    __tsan_acquire(ptr);
+#endif
+}
+
+void utils_annotate_release(void *ptr) {
+#if __SANITIZE_THREAD__
+    __tsan_release(ptr);
+#endif
+}
+
+#ifdef __cplusplus
+}
+#endif