From 47a351c67b79cbdb7695ea712d3683caa9a0f2d2 Mon Sep 17 00:00:00 2001 From: guru4elephant Date: Tue, 18 Feb 2020 21:08:35 +0800 Subject: [PATCH] add profile server argument on request, if profile_server=true, timestamp of server will be returned --- core/general-server/proto/general_model_service.proto | 1 + core/sdk-cpp/proto/general_model_service.proto | 1 + python/paddle_serving_client/__init__.py | 4 ++++ 3 files changed, 6 insertions(+) diff --git a/core/general-server/proto/general_model_service.proto b/core/general-server/proto/general_model_service.proto index 06b08438..9ffc5df7 100644 --- a/core/general-server/proto/general_model_service.proto +++ b/core/general-server/proto/general_model_service.proto @@ -38,6 +38,7 @@ message FetchInst { message Request { repeated FeedInst insts = 1; repeated string fetch_var_names = 2; + optional bool profile_server = 3 [ default = false ]; }; message Response { diff --git a/core/sdk-cpp/proto/general_model_service.proto b/core/sdk-cpp/proto/general_model_service.proto index 28f17d5e..ec942706 100644 --- a/core/sdk-cpp/proto/general_model_service.proto +++ b/core/sdk-cpp/proto/general_model_service.proto @@ -38,6 +38,7 @@ message FetchInst { message Request { repeated FeedInst insts = 1; repeated string fetch_var_names = 2; + optional bool profile_server = 3 [ default = false ]; }; message Response { diff --git a/python/paddle_serving_client/__init__.py b/python/paddle_serving_client/__init__.py index 0b2433d0..79bf945d 100644 --- a/python/paddle_serving_client/__init__.py +++ b/python/paddle_serving_client/__init__.py @@ -17,6 +17,7 @@ from .proto import sdk_configure_pb2 as sdk from .proto import general_model_config_pb2 as m_config import google.protobuf.text_format import time +import sys int_type = 0 float_type = 1 @@ -87,6 +88,9 @@ class Client(object): # map feed names to index self.client_handle_ = PredictorClient() self.client_handle_.init(path) + read_env_flags = ["profile_client", "profile_server"] + self.client_handle_.init_gflags([sys.argv[0]] + + ["--tryfromenv=" + ",".join(read_env_flags)]) self.feed_names_ = [var.alias_name for var in model_conf.feed_var] self.fetch_names_ = [var.alias_name for var in model_conf.fetch_var] self.feed_shapes_ = [var.shape for var in model_conf.feed_var] -- GitLab