PaddlePredictorImpl::Clone return unique_ptr

7 years ago · 47d9f64ff7
parent ad0bebd515
commit 47d9f64ff7
1 changed files with 3 additions and 2 deletions
--- a/paddle/contrib/inference/paddle_inference_api_impl.cc
+++ b/paddle/contrib/inference/paddle_inference_api_impl.cc
@ -135,10 +135,11 @@ bool PaddlePredictorImpl::Run(const std::vector<PaddleTensor> &inputs,

 std::unique_ptr<PaddlePredictor> PaddlePredictorImpl::Clone() {
  VLOG(3) << "Predictor::clone";
-  std::unique_ptr<PaddlePredictor> cls(new PaddlePredictorImpl(config_));
+  std::unique_ptr<PaddlePredictor> cls(nullptr);
  if (!cls->InitShared()) {
    LOG(ERROR) << "fail to call InitShared";
-    return nullptr;
+  } else {
+    cls.reset(new PaddlePredictorImpl(config_));
  }
  // fix manylinux compile error.
  return cls;