From 67b59b710b67a106620e5e9a65fdd8a903fc65d6 Mon Sep 17 00:00:00 2001
From: MRXLT <xlt2024@gmail.com>
Date: Fri, 3 Apr 2020 18:28:01 +0800
Subject: [PATCH] fix gpu server log

---
 core/predictor/framework/infer.h | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/core/predictor/framework/infer.h b/core/predictor/framework/infer.h
index 177b6274..f8bd7843 100644
--- a/core/predictor/framework/infer.h
+++ b/core/predictor/framework/infer.h
@@ -17,6 +17,7 @@
 #include <sys/types.h>
 #include <unistd.h>
 #include <string>
+#include <utility>
 #include <vector>
 #include "core/predictor/common/inner_common.h"
 #include "core/predictor/framework/bsf.h"
@@ -588,12 +589,14 @@ class VersionedInferEngine : public InferEngine {
       LOG(ERROR) << "Failed generate engine with type:" << engine_type;
       return -1;
     }
-
+    VLOG(2) << "FLGS_logtostderr " << FLAGS_logtostderr;
+    int tmp = FLAGS_logtostderr;
     if (engine->proc_initialize(conf, version) != 0) {
       LOG(ERROR) << "Failed initialize engine, type:" << engine_type;
       return -1;
     }
-
+    VLOG(2) << "FLGS_logtostderr " << FLAGS_logtostderr;
+    FLAGS_logtostderr = tmp;
     auto r = _versions.insert(std::make_pair(engine->version(), engine));
     if (!r.second) {
       LOG(ERROR) << "Failed insert item: " << engine->version()
@@ -760,7 +763,6 @@ class InferManager {
       LOG(ERROR) << "failed load infer config, path: " << path << "/" << file;
       return -1;
     }
-
     size_t engine_num = model_toolkit_conf.engines_size();
     for (size_t ei = 0; ei < engine_num; ++ei) {
       std::string engine_name = model_toolkit_conf.engines(ei).name();
@@ -769,12 +771,10 @@ class InferManager {
         LOG(ERROR) << "Failed generate versioned engine: " << engine_name;
         return -1;
       }
-
       if (engine->proc_initialize(model_toolkit_conf.engines(ei)) != 0) {
         LOG(ERROR) << "Failed initialize version engine, name:" << engine_name;
         return -1;
       }
-
       auto r = _map.insert(std::make_pair(engine_name, engine));
       if (!r.second) {
         LOG(ERROR) << "Failed insert item: " << engine_name;
@@ -782,7 +782,6 @@ class InferManager {
       }
       LOG(WARNING) << "Succ proc initialize engine: " << engine_name;
     }
-
     return 0;
   }
 
-- 
GitLab