portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-11-03 21:44:33 +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 cc6218a692
commit 8d1b26dd4b
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
lib/Grid_aligned_allocator.h
View File

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