提交 e4b2ff8b 编写于 作者: B barrierye

fix engine_name in cube_infer_op

上级 1a790771
...@@ -158,8 +158,8 @@ int GeneralDistKVInferOp::inference() { ...@@ -158,8 +158,8 @@ int GeneralDistKVInferOp::inference() {
timeline.Start(); timeline.Start();
if (InferManager::instance().infer( if (InferManager::instance().infer(
GENERAL_MODEL_NAME, &infer_in, out, batch_size)) { engine_name().c_str(), &infer_in, out, batch_size)) {
LOG(ERROR) << "Failed do infer in fluid model: " << GENERAL_MODEL_NAME; LOG(ERROR) << "Failed do infer in fluid model: " << engine_name();
return -1; return -1;
} }
......
...@@ -189,8 +189,8 @@ int GeneralDistKVQuantInferOp::inference() { ...@@ -189,8 +189,8 @@ int GeneralDistKVQuantInferOp::inference() {
timeline.Start(); timeline.Start();
if (InferManager::instance().infer( if (InferManager::instance().infer(
GENERAL_MODEL_NAME, &infer_in, out, batch_size)) { engine_name().c_str(), &infer_in, out, batch_size)) {
LOG(ERROR) << "Failed do infer in fluid model: " << GENERAL_MODEL_NAME; LOG(ERROR) << "Failed do infer in fluid model: " << engine_name();
return -1; return -1;
} }
......
...@@ -31,8 +31,6 @@ namespace baidu { ...@@ -31,8 +31,6 @@ namespace baidu {
namespace paddle_serving { namespace paddle_serving {
namespace serving { namespace serving {
static const char* GENERAL_MODEL_NAME = "general_model";
struct GeneralBlob { struct GeneralBlob {
std::vector<paddle::PaddleTensor> tensor_vector; std::vector<paddle::PaddleTensor> tensor_vector;
int64_t time_stamp[20]; int64_t time_stamp[20];
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册