1 /*===---- mm_malloc.h - Allocating and Freeing Aligned Memory Blocks -------=== 2 * 3 * Permission is hereby granted, free of charge, to any person obtaining a copy 4 * of this software and associated documentation files (the "Software"), to deal 5 * in the Software without restriction, including without limitation the rights 6 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 7 * copies of the Software, and to permit persons to whom the Software is 8 * furnished to do so, subject to the following conditions: 9 * 10 * The above copyright notice and this permission notice shall be included in 11 * all copies or substantial portions of the Software. 12 * 13 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 14 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 15 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 16 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 17 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 18 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 19 * THE SOFTWARE. 20 * 21 *===-----------------------------------------------------------------------=== 22 */ 23 24 #ifndef __MM_MALLOC_H 25 #define __MM_MALLOC_H 26 27 #include <stdlib.h> 28 29 #ifdef _WIN32 30 #include <malloc.h> 31 #else 32 #ifndef __cplusplus 33 extern int posix_memalign(void **__memptr, size_t __alignment, size_t __size); 34 #else 35 // Some systems (e.g. those with GNU libc) declare posix_memalign with an 36 // exception specifier. Via an "egregious workaround" in 37 // Sema::CheckEquivalentExceptionSpec, Clang accepts the following as a valid 38 // redeclaration of glibc's declaration. 39 extern "C" int posix_memalign(void **__memptr, size_t __alignment, size_t __size); 40 #endif 41 #endif 42 43 #if !(defined(_WIN32) && defined(_mm_malloc)) 44 static __inline__ void *__attribute__((__always_inline__, __nodebug__, 45 __malloc__)) 46 _mm_malloc(size_t __size, size_t __align) 47 { 48 if (__align == 1) { 49 return malloc(__size); 50 } 51 52 if (!(__align & (__align - 1)) && __align < sizeof(void *)) 53 __align = sizeof(void *); 54 55 void *__mallocedMemory; 56 #if defined(__MINGW32__) 57 __mallocedMemory = __mingw_aligned_malloc(__size, __align); 58 #elif defined(_WIN32) 59 __mallocedMemory = _aligned_malloc(__size, __align); 60 #else 61 if (posix_memalign(&__mallocedMemory, __align, __size)) 62 return 0; 63 #endif 64 65 return __mallocedMemory; 66 } 67 68 static __inline__ void __attribute__((__always_inline__, __nodebug__)) 69 _mm_free(void *__p) 70 { 71 free(__p); 72 } 73 #endif 74 75 #endif /* __MM_MALLOC_H */ 76
