Refine code.

paddle/fluid/framework/init.cc

@@ -20,7 +20,6 @@ limitations under the License. */
 #include "paddle/fluid/framework/init.h"
 #include "paddle/fluid/framework/operator.h"
 #include "paddle/fluid/platform/device_context.h"
-#include "paddle/fluid/platform/device_context.h"
 #include "paddle/fluid/platform/place.h"
 #include "paddle/fluid/string/piece.h"
 
@@ -35,10 +34,8 @@ std::once_flag p2p_init_flag;
 
 using paddle::platform::DeviceContextPool;
 
-void Init(int argc, char **argv) {
-  std::call_once(gflags_init_flag,
-                 [&]() { google::ParseCommandLineFlags(&argc, &argv, true); });
-
+void Init(std::vector<std::string> &argv) {
+  InitGflags(argv);
   // init devices
   std::vector<int> devices;
   std::string token;
@@ -51,6 +48,7 @@ void Init(int argc, char **argv) {
 
 void InitGflags(std::vector<std::string> &argv) {
   std::call_once(gflags_init_flag, [&]() {
+    argv.push_back("dummy");
     int argc = argv.size();
     char **arr = new char *[argv.size()];
     std::string line;
@@ -151,7 +149,7 @@ void InitDevices(bool init_p2p, const std::vector<int> devices) {
 #endif
 
   for (size_t i = 0; i < devices.size(); ++i) {
-    if (devices[i] >= count) {
+    if (devices[i] >= count || devices[i] < 0) {
       LOG(WARNING) << "Invalid devices id.";
       continue;
     }

paddle/fluid/framework/init.h

@@ -20,7 +20,7 @@ limitations under the License. */
 namespace paddle {
 namespace framework {
 
-void Init(int argc, char **argv);
+void Init(std::vector<std::string> &argv);
 
 void InitGflags(std::vector<std::string> &argv);
 

paddle/fluid/inference/io.cc

@@ -24,11 +24,7 @@ limitations under the License. */
 namespace paddle {
 namespace inference {
 
-// Temporarily add this function for exposing framework::InitDevices() when
-// linking the inference shared library.
-void Init(bool init_p2p) { framework::InitDevices(init_p2p); }
-
-void Init(int argc, char** argv) { framework::Init(argc, argv); }
+void Init(std::vector<std::string> &argv) { framework::Init(argv); }
 
 void ReadBinaryFile(const std::string& filename, std::string* contents) {
   std::ifstream fin(filename, std::ios::in | std::ios::binary);

paddle/fluid/inference/io.h

@@ -25,9 +25,7 @@ limitations under the License. */
 namespace paddle {
 namespace inference {
 
-void Init(bool init_p2p);
-
-void Init(int argc, char** argv);
+void Init(std::vector<std::string> &argv);
 
 void LoadPersistables(framework::Executor* executor, framework::Scope* scope,
                       const framework::ProgramDesc& main_program,