mm_malloc.h (1929B) - Raw
1 /*===---- mm_malloc.h - Allocating and Freeing Aligned Memory Blocks -------=== 2 * 3 * Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 * See https://llvm.org/LICENSE.txt for license information. 5 * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 * 7 *===-----------------------------------------------------------------------=== 8 */ 9 10 #ifndef __MM_MALLOC_H 11 #define __MM_MALLOC_H 12 13 #include <stdlib.h> 14 15 #ifdef _WIN32 16 #include <malloc.h> 17 #else 18 #ifndef __cplusplus 19 extern int posix_memalign(void **__memptr, size_t __alignment, size_t __size); 20 #else 21 // Some systems (e.g. those with GNU libc) declare posix_memalign with an 22 // exception specifier. Via an "egregious workaround" in 23 // Sema::CheckEquivalentExceptionSpec, Clang accepts the following as a valid 24 // redeclaration of glibc's declaration. 25 extern "C" int posix_memalign(void **__memptr, size_t __alignment, size_t __size); 26 #endif 27 #endif 28 29 #if !(defined(_WIN32) && defined(_mm_malloc)) 30 static __inline__ void *__attribute__((__always_inline__, __nodebug__, 31 __malloc__, __alloc_size__(1), 32 __alloc_align__(2))) 33 _mm_malloc(size_t __size, size_t __align) { 34 if (__align == 1) { 35 return malloc(__size); 36 } 37 38 if (!(__align & (__align - 1)) && __align < sizeof(void *)) 39 __align = sizeof(void *); 40 41 void *__mallocedMemory; 42 #if defined(__MINGW32__) 43 __mallocedMemory = __mingw_aligned_malloc(__size, __align); 44 #elif defined(_WIN32) 45 __mallocedMemory = _aligned_malloc(__size, __align); 46 #else 47 if (posix_memalign(&__mallocedMemory, __align, __size)) 48 return 0; 49 #endif 50 51 return __mallocedMemory; 52 } 53 54 static __inline__ void __attribute__((__always_inline__, __nodebug__)) 55 _mm_free(void *__p) 56 { 57 #if defined(__MINGW32__) 58 __mingw_aligned_free(__p); 59 #elif defined(_WIN32) 60 _aligned_free(__p); 61 #else 62 free(__p); 63 #endif 64 } 65 #endif 66 67 #endif /* __MM_MALLOC_H */