portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-10-30 19:44:32 +00:00
Code Releases Activity

Key of mm_malloc.h

Browse Source
This commit is contained in:
Peter Boyle
2015-05-15 11:32:11 +01:00
parent c0977dcfaa
commit a36c974f26
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
lib/Grid_aligned_allocator.h
View File

@@ -2,6 +2,9 @@
#define GRID_ALIGNED_ALLOCATOR_H #define GRID_ALIGNED_ALLOCATOR_H
#include <immintrin.h> #include <immintrin.h>
#ifdef HAVE_MM_MALLOC_H
#include <mm_malloc.h>
#endif
namespace Grid { namespace Grid {
@@ -36,16 +39,20 @@ public:
pointer allocate(size_type __n, const void* = 0) pointer allocate(size_type __n, const void* = 0)
{ {
#ifdef AVX512 #ifdef HAVE_MM_MALLOC_H
_Tp * ptr = (_Tp *) memalign(128,__n*sizeof(_Tp));
#else
_Tp * ptr = (_Tp *) _mm_malloc(__n*sizeof(_Tp),128); _Tp * ptr = (_Tp *) _mm_malloc(__n*sizeof(_Tp),128);
#else
_Tp * ptr = (_Tp *) memalign(128,__n*sizeof(_Tp));
#endif #endif
return ptr; return ptr;
} }
void deallocate(pointer __p, size_type) { void deallocate(pointer __p, size_type) {
#ifdef HAVE_MM_MALLOC_H
_mm_free(__p);
#else
free(__p); free(__p);
#endif
} }
void construct(pointer __p, const _Tp& __val) { }; void construct(pointer __p, const _Tp& __val) { };
void construct(pointer __p) { }; void construct(pointer __p) { };