OLD | NEW |
| (Empty) |
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 */ | |
OLD | NEW |