From 7ffa04fff692e755d13a1573bb05849e6d40e637 Mon Sep 17 00:00:00 2001
From: wangguibao <wangguibao@baidu.com>
Date: Tue, 21 May 2019 15:34:34 +0800
Subject: [PATCH] Fix image classification bug

---
 demo-serving/op/reader_op.cpp | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/demo-serving/op/reader_op.cpp b/demo-serving/op/reader_op.cpp
index d50c9ebd..c57e1519 100644
--- a/demo-serving/op/reader_op.cpp
+++ b/demo-serving/op/reader_op.cpp
@@ -63,9 +63,11 @@ int ReaderOp::inference() {
   // tls resource assignment
   size_t dense_capacity = 3 * resize.width * resize.height;
   size_t len = dense_capacity * sizeof(float) * sample_size;
-  float* data =
-      reinterpret_cast<float*>(MempoolWrapper::instance().malloc(len));
-  if (data == NULL) {
+
+  // Allocate buffer in PaddleTensor, so that buffer will be managed by the Tensor
+  in_tensor.data.Resize(len);
+  float *data = reinterpret_cast<float *>(in_tensor.data.data());
+  if (in_tensor.data.data() == NULL) {
     LOG(ERROR) << "Failed create temp float array, "
                << "size=" << dense_capacity * sample_size * sizeof(float);
     return -1;
@@ -144,9 +146,6 @@ int ReaderOp::inference() {
       }
     }
   }
-  paddle::PaddleBuf pbuf(data, len);
-  in_tensor.data = pbuf;
-
   in->push_back(in_tensor);
 
   return 0;
-- 
GitLab