mm_malloc.h 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  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. #ifndef __MM_MALLOC_H
  24. #define __MM_MALLOC_H
  25. #include <stdlib.h>
  26. #ifdef _WIN32
  27. #include <malloc.h>
  28. #else
  29. #ifndef __cplusplus
  30. extern int posix_memalign(void **__memptr, size_t __alignment, size_t __size);
  31. #else
  32. // Some systems (e.g. those with GNU libc) declare posix_memalign with an
  33. // exception specifier. Via an "egregious workaround" in
  34. // Sema::CheckEquivalentExceptionSpec, Clang accepts the following as a valid
  35. // redeclaration of glibc's declaration.
  36. extern "C" int posix_memalign(void **__memptr, size_t __alignment, size_t __size);
  37. #endif
  38. #endif
  39. #if !(defined(_WIN32) && defined(_mm_malloc))
  40. static __inline__ void *__attribute__((__always_inline__, __nodebug__,
  41. __malloc__))
  42. _mm_malloc(size_t __size, size_t __align)
  43. {
  44. if (__align == 1) {
  45. return malloc(__size);
  46. }
  47. if (!(__align & (__align - 1)) && __align < sizeof(void *))
  48. __align = sizeof(void *);
  49. void *__mallocedMemory;
  50. #if defined(__MINGW32__)
  51. __mallocedMemory = __mingw_aligned_malloc(__size, __align);
  52. #elif defined(_WIN32)
  53. __mallocedMemory = _aligned_malloc(__size, __align);
  54. #else
  55. if (posix_memalign(&__mallocedMemory, __align, __size))
  56. return 0;
  57. #endif
  58. return __mallocedMemory;
  59. }
  60. static __inline__ void __attribute__((__always_inline__, __nodebug__))
  61. _mm_free(void *__p)
  62. {
  63. free(__p);
  64. }
  65. #endif
  66. #endif /* __MM_MALLOC_H */