add workspace

lite/api/paddle_api.cc

@@ -237,6 +237,10 @@ void PaddlePredictor::SaveOptimizedModel(const std::string &model_dir,
       << "The SaveOptimizedModel API is only supported by CxxConfig predictor.";
 }
 
+PaddlePredictor::~PaddlePredictor() {
+  lite::DeviceInfo::Global().CleanWorkspace();
+}
+
 template <typename ConfigT>
 std::shared_ptr<PaddlePredictor> CreatePaddlePredictor(const ConfigT &) {
   return std::shared_ptr<PaddlePredictor>();

lite/api/paddle_api.h

@@ -118,7 +118,7 @@ class LITE_API PaddlePredictor {
       LiteModelType model_type = LiteModelType::kProtobuf,
       bool record_info = false);
 
-  virtual ~PaddlePredictor() = default;
+  virtual ~PaddlePredictor();
 
  protected:
   int threads_{1};

lite/core/device_info.h

@@ -77,6 +77,11 @@ class DeviceInfo {
     return reinterpret_cast<T*>(workspace_.mutable_data<int8_t>());
   }
   bool ExtendWorkspace(size_t size);
+  bool CleanWorkspace() {
+    // workspace_.Resize({1});
+    // workspace_.mutable_data<int8_t>();
+    workspace_.clear();
+  }
 
  private:
   int core_num_;