diff --git a/core/general-server/proto/general_model_service.proto b/core/general-server/proto/general_model_service.proto index 06b08438ab648275d291778a92b73460d56058a2..9ffc5df7b550b384080c7a5372280f708e01a720 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 28f17d5e145488c6df4dd67d9b3a4d02f698a19d..ec942706acf3cabcb439e24868ac8ea61809104c 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 0b2433d0f8daf0133c417404cc0e6d9182737465..79bf945dbc8c3cac4cb51e8c1856a59b11c37ac5 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]