Warning: This file is not a C or C++ file. It does not have highlighting.
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 */ |
68 |
Warning: This file is not a C or C++ file. It does not have highlighting.