Merge pull request #5679 from hedaoyuan/inference

Fix crashes due to multiple calls to paddle_init
PaddlePaddle · Nov 16, 2017 · 08bc08d · 08bc08d
2 parents 6cf7f1e + 6303d3b
commit 08bc08d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/paddle/capi/Main.cpp b/paddle/capi/Main.cpp
@@ -29,6 +29,9 @@ static void initPaddle(int argc, char** argv) {
 
 extern "C" {
 paddle_error paddle_init(int argc, char** argv) {
+  static bool isInit = false;
+  if (isInit) return kPD_NO_ERROR;
+
   std::vector<char*> realArgv;
   realArgv.reserve(argc + 1);
   realArgv.push_back(strdup(""));
@@ -37,6 +40,7 @@ paddle_error paddle_init(int argc, char** argv) {
   }
   initPaddle(argc + 1, realArgv.data());
   free(realArgv[0]);
+  isInit = true;
   return kPD_NO_ERROR;
 }
 }