1 /* 2 This file is part of the BOLT-LMM linear mixed model software package 3 developed by Po-Ru Loh. Copyright (C) 2014-2019 Harvard University. 4 5 This program is free software: you can redistribute it and/or modify 6 it under the terms of the GNU General Public License as published by 7 the Free Software Foundation, either version 3 of the License, or 8 (at your option) any later version. 9 10 This program is distributed in the hope that it will be useful, 11 but WITHOUT ANY WARRANTY; without even the implied warranty of 12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 13 GNU General Public License for more details. 14 15 You should have received a copy of the GNU General Public License 16 along with this program. If not, see <http://www.gnu.org/licenses/>. 17 */ 18 19 #ifndef MEMORYUTILS_HPP 20 #define MEMORYUTILS_HPP 21 22 #include "Types.hpp" 23 24 #define MEM_ALIGNMENT 64 25 26 //#define ALIGNED_MALLOC(size) mkl_malloc(size, MEM_ALIGNMENT) 27 //#define ALIGNED_MALLOC(size) _mm_malloc(size, MEM_ALIGNMENT) 28 void *ALIGNED_MALLOC(uint64 size); 29 30 #ifdef USE_MKL_MALLOC 31 #include <mkl.h> 32 #define ALIGNED_FREE mkl_free 33 #else 34 #include <xmmintrin.h> 35 #define ALIGNED_FREE _mm_free 36 #endif 37 38 #define ALIGNED_MALLOC_DOUBLES(numDoubles) (double *) ALIGNED_MALLOC((numDoubles)*sizeof(double)) 39 #define ALIGNED_MALLOC_FLOATS(numFloats) (float *) ALIGNED_MALLOC((numFloats)*sizeof(float)) 40 #define ALIGNED_MALLOC_UCHARS(numUchars) (uchar *) ALIGNED_MALLOC((numUchars)*sizeof(uchar)) 41 42 #endif 43