|
|
@ -43,14 +43,16 @@ void* CPUAllocator::Alloc(size_t* index, size_t size) {
|
|
|
|
|
|
|
|
|
|
|
|
*index = 0; // unlock memory
|
|
|
|
*index = 0; // unlock memory
|
|
|
|
|
|
|
|
|
|
|
|
void* p;
|
|
|
|
void* p = nullptr;
|
|
|
|
|
|
|
|
|
|
|
|
#ifdef PADDLE_WITH_MKLDNN
|
|
|
|
#ifdef PADDLE_WITH_MKLDNN
|
|
|
|
// refer to https://github.com/01org/mkl-dnn/blob/master/include/mkldnn.hpp
|
|
|
|
// refer to https://github.com/01org/mkl-dnn/blob/master/include/mkldnn.hpp
|
|
|
|
// memory alignment
|
|
|
|
// memory alignment
|
|
|
|
PADDLE_ENFORCE_EQ(posix_memalign(&p, 4096ul, size), 0);
|
|
|
|
PADDLE_ENFORCE_EQ(posix_memalign(&p, 4096ul, size), 0, "Alloc %ld error!",
|
|
|
|
|
|
|
|
size);
|
|
|
|
#else
|
|
|
|
#else
|
|
|
|
PADDLE_ENFORCE_EQ(posix_memalign(&p, 32ul, size), 0);
|
|
|
|
PADDLE_ENFORCE_EQ(posix_memalign(&p, 32ul, size), 0, "Alloc %ld error!",
|
|
|
|
|
|
|
|
size);
|
|
|
|
#endif
|
|
|
|
#endif
|
|
|
|
PADDLE_ENFORCE(p, "Fail to allocate CPU memory: size = %d .", size);
|
|
|
|
PADDLE_ENFORCE(p, "Fail to allocate CPU memory: size = %d .", size);
|
|
|
|
|
|
|
|
|
|
|
|