From ed9fa4b3011bd5e092819776a55922aac8ce9fdf Mon Sep 17 00:00:00 2001 From: Qiao Longfei Date: Mon, 26 Nov 2018 16:17:51 +0800 Subject: [PATCH] can run --- paddle/fluid/operators/distributed/grpc_server.cc | 2 +- paddle/fluid/operators/distributed/parameter_prefetch.cc | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/paddle/fluid/operators/distributed/grpc_server.cc b/paddle/fluid/operators/distributed/grpc_server.cc index d5295dc63..ad74e7a03 100644 --- a/paddle/fluid/operators/distributed/grpc_server.cc +++ b/paddle/fluid/operators/distributed/grpc_server.cc @@ -192,7 +192,7 @@ class RequestPrefetch final : public RequestBase { framework::Variable* outvar = scope->Var(out_var_name); request_handler_->Handle(in_var_name, scope, invar, &outvar, trainer_id, - out_var_name); + out_var_name, table_name); SerializeToByteBuffer(out_var_name, outvar, *request_handler_->dev_ctx(), &reply_); diff --git a/paddle/fluid/operators/distributed/parameter_prefetch.cc b/paddle/fluid/operators/distributed/parameter_prefetch.cc index 4d677e30b..23beabc6e 100644 --- a/paddle/fluid/operators/distributed/parameter_prefetch.cc +++ b/paddle/fluid/operators/distributed/parameter_prefetch.cc @@ -106,7 +106,7 @@ inline void MergeMultipleVarsIntoOnBySection( const std::vector& height_section, const std::vector>& splited_ids, const framework::ExecutionContext& context, framework::Scope* scope) { - PADDLE_ENFORCE_EQ(out_var_names.size(), height_section.size() + 1, ""); + PADDLE_ENFORCE_EQ(out_var_names.size(), height_section.size(), ""); auto cpu_place = platform::CPUPlace(); @@ -185,7 +185,7 @@ void prefetch(const std::string& id_name, const std::string& out_name, << " to get " << out_var_names[i] << " back"; rets.push_back(rpc_client->AsyncPrefetchVar( epmap[i], ctx, local_scope, in_var_names[i], out_var_names[i], - table_name)); + table_name + ".block" + std::to_string(i))); } else { VLOG(30) << "don't send no-initialied variable: " << out_var_names[i]; } -- GitLab