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

Loading…
Cancel
Save