Disable memory poisoning in DisjointPool

ASan throws an error whenever the memory poison is called for the memory allocated on GPU:
"AddressSanitizer: CHECK failed: asan_mapping.h:359 "((AddrIsInMem(p))) != (0)" (0x0, 0x0)".

This commit disables poisoning.

Author: PatKamin

src/pool/pool_disjoint.cpp

@@ -31,6 +31,11 @@
 #include "utils_math.h"
 #include "utils_sanitizers.h"
 
+// Temporary solution for disabling memory poisoning. This is needed because
+// AddressSanitizer does not support memory poisoning for GPU allocations.
+// More info: https://github.com/oneapi-src/unified-memory-framework/issues/634
+#define POISON_MEMORY 0
+
 typedef struct umf_disjoint_pool_shared_limits_t {
     size_t MaxSize;
     std::atomic<size_t> TotalSize;
@@ -400,7 +405,9 @@ static void *memoryProviderAlloc(umf_memory_provider_handle_t hProvider,
     if (ret != UMF_RESULT_SUCCESS) {
         throw MemoryProviderError{ret};
     }
+#if POISON_MEMORY
     utils_annotate_memory_inaccessible(ptr, size);
+#endif
     return ptr;
 }
 
@@ -822,7 +829,9 @@ void *DisjointPool::AllocImpl::allocate(size_t Size, bool &FromPool) try {
     FromPool = false;
     if (Size > getParams().MaxPoolableSize) {
         Ptr = memoryProviderAlloc(getMemHandle(), Size);
+#if POISON_MEMORY
         utils_annotate_memory_undefined(Ptr, Size);
+#endif
         return Ptr;
     }
 
@@ -839,7 +848,9 @@ void *DisjointPool::AllocImpl::allocate(size_t Size, bool &FromPool) try {
     }
 
     VALGRIND_DO_MEMPOOL_ALLOC(this, Ptr, Size);
+#if POISON_MEMORY
     utils_annotate_memory_undefined(Ptr, Bucket.getSize());
+#endif
 
     return Ptr;
 } catch (MemoryProviderError &e) {
@@ -877,7 +888,9 @@ void *DisjointPool::AllocImpl::allocate(size_t Size, size_t Alignment,
     FromPool = false;
     if (AlignedSize > getParams().MaxPoolableSize) {
         Ptr = memoryProviderAlloc(getMemHandle(), Size, Alignment);
+#if POISON_MEMORY
         utils_annotate_memory_undefined(Ptr, Size);
+#endif
         return Ptr;
     }
 
@@ -894,8 +907,9 @@ void *DisjointPool::AllocImpl::allocate(size_t Size, size_t Alignment,
     }
 
     VALGRIND_DO_MEMPOOL_ALLOC(this, AlignPtrUp(Ptr, Alignment), Size);
+#if POISON_MEMORY
     utils_annotate_memory_undefined(AlignPtrUp(Ptr, Alignment), Size);
-
+#endif
     return AlignPtrUp(Ptr, Alignment);
 } catch (MemoryProviderError &e) {
     umf::getPoolLastStatusRef<DisjointPool>() = e.code;
@@ -962,8 +976,9 @@ void DisjointPool::AllocImpl::deallocate(void *Ptr, bool &ToPool) {
             }
 
             VALGRIND_DO_MEMPOOL_FREE(this, Ptr);
+#if POISON_MEMORY
             utils_annotate_memory_inaccessible(Ptr, Bucket.getSize());
-
+#endif
             if (Bucket.getSize() <= Bucket.ChunkCutOff()) {
                 Bucket.freeChunk(Ptr, Slab, ToPool);
             } else {