annotate alloc on unixes. (#548)

* annotate `alloc` on unixes. * changes from feedback
microsoft · Sep 25, 2022 · a060462 · a060462
1 parent 591dc4c
commit a060462
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/snmalloc/mem/localalloc.h b/src/snmalloc/mem/localalloc.h
@@ -1,7 +1,9 @@
 #pragma once
 
-#ifdef _MSC_VER
-#  define ALLOCATOR __declspec(allocator)
+#if defined(_MSC_VER)
+#  define ALLOCATOR __declspec(allocator) __declspec(restrict)
+#elif __has_attribute(malloc)
+#  define ALLOCATOR __attribute__((malloc))
 #else
 #  define ALLOCATOR
 #endif