Merge pull request #1403 from ShiningZhang/dev-bug

fix bug: client core while low version server do not have profile_time

Merge pull request #1403 from ShiningZhang/dev-bug
fix bug: client core while low version server do not have profile_time
99944bda · TeslaZhao · GitHub · eb16b33f · 4a7a5755 · 99944bda
隐藏空白更改
内联并排

Showing with 12 addition and 8 deletion

core/general-client/src/brpc_client.cpp core/general-client/src/brpc_client.cpp +6 -4

core/general-client/src/general_model.cpp core/general-client/src/general_model.cpp +6 -4

未找到文件。
--- a/core/general-client/src/brpc_client.cpp
+++ b/core/general-client/src/brpc_client.cpp
@@ -184,10 +184,12 @@ int ServingBrpcClient::predict(const PredictorInputs& inputs,
      oss << "op" << i << "=" << t << "ms,";
    }
  }
-  int i = op_num - 1;
+  if (op_num > 0) {
-  double server_cost = (res.profile_time(i * 2 + 1)
+    int i = op_num - 1;
-               - res.profile_time(i * 2)) / 1000.0;
+    double server_cost = (res.profile_time(i * 2 + 1)
-  oss << "server_cost=" << server_cost << "ms.";
+                 - res.profile_time(i * 2)) / 1000.0;
+    oss << "server_cost=" << server_cost << "ms.";
+  }
  LOG(INFO) << oss.str();
  return 0;

--- a/core/general-client/src/general_model.cpp
+++ b/core/general-client/src/general_model.cpp
@@ -449,10 +449,12 @@ int PredictorClient::numpy_predict(
      oss << "op" << i << "=" << t << "ms,";
    }
  }
-  int i = op_num - 1;
+  if (op_num > 0) {
-  double server_cost = (res.profile_time(i * 2 + 1)
+    int i = op_num - 1;
-               - res.profile_time(i * 2)) / 1000.0;
+    double server_cost = (res.profile_time(i * 2 + 1)
-  oss << "server_cost=" << server_cost << "ms.";
+                 - res.profile_time(i * 2)) / 1000.0;
+    oss << "server_cost=" << server_cost << "ms.";
+  }
  LOG(INFO) << oss.str();
  return 0;
 }