marian-nmt/marian
1
1
Fork 0
mirror of https://github.com/marian-nmt/marian.git synced 2024-11-30 12:02:16 +03:00
Code Issues Packages Projects Releases Wiki Activity

add _aligned_free for windows

Browse Source
This commit is contained in:
Marcin Junczys-Dowmunt 2018-12-12 15:08:55 -08:00
parent 28629ea516
commit 19cfb307c4
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/tensors/cpu/device.cpp
View File

@ -29,6 +29,12 @@ Device::~Device() {
#define MALLOC(size) malloc(size)
#endif
#ifdef _WIN32
#define FREE(ptr) _aligned_free(ptr)
#else
#define FREE(ptr) free(ptr)
#endif
void Device::reserve(size_t size) {
size = align(size);
ABORT_IF(size < size_ || size == 0,
@ -37,7 +43,7 @@ void Device::reserve(size_t size) {
if(data_) {
uint8_t *temp = static_cast<uint8_t*>(MALLOC(size));
std::copy(data_, data_ + size_, temp);
free(data_);
FREE(data_);
data_ = temp;
} else {
data_ = static_cast<uint8_t*>(MALLOC(size));