From 6aadcac74405e02586d485de1458a1ad3b5c688f Mon Sep 17 00:00:00 2001 From: wangguibao Date: Thu, 1 Aug 2019 16:54:08 +0800 Subject: [PATCH] CTR prediction serving --- demo-serving/op/ctr_prediction_op.cpp | 2 +- predictor/framework/kv_manager.h | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/demo-serving/op/ctr_prediction_op.cpp b/demo-serving/op/ctr_prediction_op.cpp index 4311148e..8e0e6fc0 100644 --- a/demo-serving/op/ctr_prediction_op.cpp +++ b/demo-serving/op/ctr_prediction_op.cpp @@ -109,7 +109,7 @@ int CTRPredictionOp::inference() { if (kvinfo != NULL) { std::string table_name; if (kvinfo->sparse_param_service_type != configure::EngineDesc::NONE) { - std::string table_name = kvinfo->sparse_param_service_table_name; + table_name = kvinfo->sparse_param_service_table_name; } if (kvinfo->sparse_param_service_type == configure::EngineDesc::LOCAL) { diff --git a/predictor/framework/kv_manager.h b/predictor/framework/kv_manager.h index 35031f96..c1ca8409 100644 --- a/predictor/framework/kv_manager.h +++ b/predictor/framework/kv_manager.h @@ -66,6 +66,9 @@ class KVManager { LOG(ERROR) << "Failed insert item: " << engine_name; return -1; } + + LOG(WARNING) << engine_name << ": " + << kvinfo->sparse_param_service_table_name; LOG(WARNING) << "Succ proc initialize kvmanager for engine: " << engine_name; } -- GitLab