Make api.cc malloc consistent with paddle_api.h for PaddleBuf (#22255)

revert-22710-feature/integrated_ps_api
liu zhengxi 5 years ago committed by GitHub
parent 4f1da4adcb
commit 07afc29e90
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -85,7 +85,7 @@ void PaddleBuf::Resize(size_t length) {
if (length_ >= length) return; if (length_ >= length) return;
if (memory_owned_) { if (memory_owned_) {
Free(); Free();
data_ = malloc(length); data_ = new char[length];
length_ = length; length_ = length;
memory_owned_ = true; memory_owned_ = true;
} else { } else {
@ -103,7 +103,7 @@ void PaddleBuf::Reset(void *data, size_t length) {
void PaddleBuf::Free() { void PaddleBuf::Free() {
if (memory_owned_ && data_) { if (memory_owned_ && data_) {
PADDLE_ENFORCE_GT(length_, 0UL); PADDLE_ENFORCE_GT(length_, 0UL);
free(static_cast<char *>(data_)); delete[] static_cast<char *>(data_);
data_ = nullptr; data_ = nullptr;
length_ = 0; length_ = 0;
} }

Loading…
Cancel
Save