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 <stddef.h> 28 29 #ifdef _WIN32 30 #include <malloc.h> 31 #else 32 33 // Forward declare allocation functions to allow this header to parse without 34 // any system headers. 35 #ifndef __cplusplus 36 extern void free(void *ptr); 37 extern void *malloc(size_t size) __attribute__((__malloc__)); 38 extern int posix_memalign(void **memptr, size_t alignment, size_t size); 39 #else 40 // Some systems (e.g. those with GNU libc) declare some of these functions with 41 // an exception specifier. Via an "egregious workaround" in 42 // Sema::CheckEquivalentExceptionSpec, Clang accepts the following as valid 43 // redeclarations of glibc's declarations. 44 extern "C" void free(void *ptr); 45 extern "C" void *malloc(size_t size) __attribute__((__malloc__)); 46 extern "C" int posix_memalign(void **memptr, size_t alignment, size_t size); 47 #endif 48 49 #endif 50 51 static __inline__ void *__attribute__((__always_inline__, __nodebug__, 52 __malloc__)) 53 _mm_malloc(size_t size, size_t align) 54 { 55 if (align == 1) { 56 return malloc(size); 57 } 58 59 if (!(align & (align - 1)) && align < sizeof(void *)) 60 align = sizeof(void *); 61 62 void *mallocedMemory; 63 #ifdef _WIN32 64 mallocedMemory = _aligned_malloc(size, align); 65 #else 66 if (posix_memalign(&mallocedMemory, align, size)) 67 return 0; 68 #endif 69 70 return mallocedMemory; 71 } 72 73 static __inline__ void __attribute__((__always_inline__, __nodebug__)) 74 _mm_free(void *p) 75 { 76 free(p); 77 } 78 79 #endif /* __MM_MALLOC_H */ 80
