提交 6cfebcac 编写于 作者: W wangjiawei04

remove stdout

上级 ce4757b9
...@@ -47,7 +47,6 @@ int GeneralInferOp::inference() { ...@@ -47,7 +47,6 @@ int GeneralInferOp::inference() {
if (InferManager::instance().infer(engine_name().c_str())) { if (InferManager::instance().infer(engine_name().c_str())) {
return -1; return -1;
} }
std::cout << "Infer Success" << std::endl;
return 0; return 0;
} }
DEFINE_OP(GeneralInferOp); DEFINE_OP(GeneralInferOp);
......
...@@ -91,10 +91,9 @@ int GeneralReaderOp::inference() { ...@@ -91,10 +91,9 @@ int GeneralReaderOp::inference() {
capacity.resize(var_num); capacity.resize(var_num);
for (int i = 0; i < var_num; ++i) { for (int i = 0; i < var_num; ++i) {
std::string tensor_name = model_config->_feed_name[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( auto lod_tensor = InferManager::instance().GetInputHandle(
engine_name.c_str(), tensor_name.c_str()); engine_name.c_str(), tensor_name.c_str());
std::cout << "PICK lod tensor. " << std::endl;
std::vector<std::vector<size_t>> lod; std::vector<std::vector<size_t>> lod;
std::vector<int> shape; std::vector<int> shape;
// get lod info here // get lod info here
...@@ -127,8 +126,6 @@ int GeneralReaderOp::inference() { ...@@ -127,8 +126,6 @@ int GeneralReaderOp::inference() {
} }
lod_tensor->SetLoD(lod); lod_tensor->SetLoD(lod);
lod_tensor->Reshape(shape); lod_tensor->Reshape(shape);
std::cout << "FINI Set Lod and Reshape, and elem type: " << elem_type[i]
<< std::endl;
// insert data here // insert data here
if (req->insts(0).tensor_array(i).elem_type() == 0) { if (req->insts(0).tensor_array(i).elem_type() == 0) {
// TODO: Copy twice here, can optimize // TODO: Copy twice here, can optimize
...@@ -156,7 +153,6 @@ int GeneralReaderOp::inference() { ...@@ -156,7 +153,6 @@ int GeneralReaderOp::inference() {
} }
lod_tensor->CopyFromCpu(dst_ptr); lod_tensor->CopyFromCpu(dst_ptr);
} }
std::cout << "FINISH Tensor Name: " << tensor_name << std::endl;
} }
return 0; return 0;
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册