From 6cfebcac9dba2b3243580872ab839befa9833b02 Mon Sep 17 00:00:00 2001 From: wangjiawei04 Date: Tue, 5 Jan 2021 08:31:24 +0000 Subject: [PATCH] remove stdout --- core/general-server/op/general_infer_op.cpp | 1 - core/general-server/op/general_reader_op.cpp | 6 +----- 2 files changed, 1 insertion(+), 6 deletions(-) diff --git a/core/general-server/op/general_infer_op.cpp b/core/general-server/op/general_infer_op.cpp index d81c604b..5b9df806 100644 --- a/core/general-server/op/general_infer_op.cpp +++ b/core/general-server/op/general_infer_op.cpp @@ -47,7 +47,6 @@ int GeneralInferOp::inference() { if (InferManager::instance().infer(engine_name().c_str())) { return -1; } - std::cout << "Infer Success" << std::endl; return 0; } DEFINE_OP(GeneralInferOp); diff --git a/core/general-server/op/general_reader_op.cpp b/core/general-server/op/general_reader_op.cpp index c14cc2c5..24259e24 100644 --- a/core/general-server/op/general_reader_op.cpp +++ b/core/general-server/op/general_reader_op.cpp @@ -91,10 +91,9 @@ int GeneralReaderOp::inference() { capacity.resize(var_num); for (int i = 0; i < var_num; ++i) { std::string tensor_name = model_config->_feed_name[i]; - std::cout << "START Tensor Name: " << tensor_name << std::endl; + VLOG(2) << "(logid=" << log_id << ") get tensor name: " << tensor_name; auto lod_tensor = InferManager::instance().GetInputHandle( engine_name.c_str(), tensor_name.c_str()); - std::cout << "PICK lod tensor. " << std::endl; std::vector> lod; std::vector shape; // get lod info here @@ -127,8 +126,6 @@ int GeneralReaderOp::inference() { } lod_tensor->SetLoD(lod); lod_tensor->Reshape(shape); - std::cout << "FINI Set Lod and Reshape, and elem type: " << elem_type[i] - << std::endl; // insert data here if (req->insts(0).tensor_array(i).elem_type() == 0) { // TODO: Copy twice here, can optimize @@ -156,7 +153,6 @@ int GeneralReaderOp::inference() { } lod_tensor->CopyFromCpu(dst_ptr); } - std::cout << "FINISH Tensor Name: " << tensor_name << std::endl; } return 0; } -- GitLab