Home | History | Annotate | Download | only in clang-include
      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