From 8e04aaa6767d74668b6fcfca2436dc3c4fc0676d Mon Sep 17 00:00:00 2001
From: xulongteng <xulongteng@baidu.com>
Date: Wed, 16 Oct 2019 14:03:50 +0800
Subject: [PATCH] fix proto

---
 demo-serving/op/bert_service_op.h     | 2 ++
 demo-serving/proto/bert_service.proto | 4 ++--
 sdk-cpp/proto/bert_service.proto      | 9 +++++++--
 3 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/demo-serving/op/bert_service_op.h b/demo-serving/op/bert_service_op.h
index 82c36623..ddc38b56 100644
--- a/demo-serving/op/bert_service_op.h
+++ b/demo-serving/op/bert_service_op.h
@@ -25,6 +25,8 @@
 #endif
 #include "demo-serving/bert_service.pb.h"
 
+#include <sys/time.h>
+
 namespace baidu {
 namespace paddle_serving {
 namespace serving {
diff --git a/demo-serving/proto/bert_service.proto b/demo-serving/proto/bert_service.proto
index 3eba5013..4ba3c536 100644
--- a/demo-serving/proto/bert_service.proto
+++ b/demo-serving/proto/bert_service.proto
@@ -25,8 +25,8 @@ message BertReqInstance {
   repeated int64 sentence_type_ids = 2;
   repeated int64 position_ids = 3;
   repeated float input_masks = 4;
-  required int64 max_seq_len = 5;
-  required int64 emb_size = 6;
+  optional int64 max_seq_len = 5;
+  optional int64 emb_size = 6;
 };
 
 message Request { repeated BertReqInstance instances = 1; };
diff --git a/sdk-cpp/proto/bert_service.proto b/sdk-cpp/proto/bert_service.proto
index 6d130fa3..fbf07e70 100644
--- a/sdk-cpp/proto/bert_service.proto
+++ b/sdk-cpp/proto/bert_service.proto
@@ -25,7 +25,8 @@ message BertReqInstance {
   repeated int64 sentence_type_ids = 2;
   repeated int64 position_ids = 3;
   repeated float input_masks = 4;
-  required int64 max_seq_len = 5;
+  optional int64 max_seq_len = 5;
+  optional int64 emb_size = 6;
 };
 
 message Request { repeated BertReqInstance instances = 1; };
@@ -34,7 +35,11 @@ message Embedding_values { repeated float values = 1; };
 
 message BertResInstance { repeated Embedding_values instances = 1; };
 
-message Response { repeated BertResInstance instances = 1; };
+message Response {
+  repeated BertResInstance instances = 1;
+  optional int64 op_time = 2;
+  optional int64 infert_time = 3;
+};
 
 service BertService {
   rpc inference(Request) returns (Response);
-- 
GitLab