提交 92c21d6e 编写于 作者: G guru4elephant

add general copy op

上级 ff597a1d
...@@ -12,7 +12,7 @@ ...@@ -12,7 +12,7 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
#include "core/general-server/op/general_dist_kv_op.h" #include "core/general-server/op/general_copy_op.h"
#include <algorithm> #include <algorithm>
#include <iostream> #include <iostream>
#include <memory> #include <memory>
...@@ -33,7 +33,7 @@ using baidu::paddle_serving::predictor::general_model::Request; ...@@ -33,7 +33,7 @@ using baidu::paddle_serving::predictor::general_model::Request;
using baidu::paddle_serving::predictor::general_model::FeedInst; using baidu::paddle_serving::predictor::general_model::FeedInst;
using baidu::paddle_serving::predictor::PaddleGeneralModelConfig; using baidu::paddle_serving::predictor::PaddleGeneralModelConfig;
int GeneralDistKVOp::inference() { int GeneralCopyOp::inference() {
// reade request from client // reade request from client
const GeneralBlob *input_blob = get_depend_argument<GeneralBlob>(pre_name()); const GeneralBlob *input_blob = get_depend_argument<GeneralBlob>(pre_name());
VLOG(2) << "precedent name: " << pre_name(); VLOG(2) << "precedent name: " << pre_name();
...@@ -82,14 +82,15 @@ int GeneralDistKVOp::inference() { ...@@ -82,14 +82,15 @@ int GeneralDistKVOp::inference() {
timeline.Pause(); timeline.Pause();
int64_t end = timeline.TimeStampUS(); int64_t end = timeline.TimeStampUS();
res->p_size = 0; CopyBlobInfo(input_blob, res)
AddBlobInfo(res, start); AddBlobInfo(res, start);
AddBlobInfo(res, end); AddBlobInfo(res, end);
VLOG(2) << "read data from client success"; VLOG(2) << "read data from client success";
return 0; return 0;
} }
DEFINE_OP(GeneralDistKVOp);
DEFINE_OP(GeneralCopyOp);
} // namespace serving } // namespace serving
} // namespace paddle_serving } // namespace paddle_serving
} // namespace baidu } // namespace baidu
...@@ -32,12 +32,12 @@ namespace baidu { ...@@ -32,12 +32,12 @@ namespace baidu {
namespace paddle_serving { namespace paddle_serving {
namespace serving { namespace serving {
class GeneralDistKVOp : class GeneralCopyOp :
public baidu::paddle_serving::predictor::OpWithChannel<GeneralBlob> { public baidu::paddle_serving::predictor::OpWithChannel<GeneralBlob> {
public: public:
typedef std::vector<paddle::PaddleTensor> TensorVector; typedef std::vector<paddle::PaddleTensor> TensorVector;
DECLARE_OP(GeneralDistKVOp); DECLARE_OP(GeneralCopyOp);
int inference(); int inference();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册