From 69c349a0ed0adaa8c0c4d6014eaf506ddcf7d23c Mon Sep 17 00:00:00 2001 From: seiriosPlus Date: Fri, 21 Aug 2020 12:54:18 +0800 Subject: [PATCH] geo sparse init from pserver --- paddle/fluid/operators/distributed/communicator.cc | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/operators/distributed/communicator.cc b/paddle/fluid/operators/distributed/communicator.cc index 1ebb1ef0592..330a766c8cb 100644 --- a/paddle/fluid/operators/distributed/communicator.cc +++ b/paddle/fluid/operators/distributed/communicator.cc @@ -676,7 +676,8 @@ void GeoCommunicator::RecvSparse(const std::string &varname) { v_delta.resize(numel); auto cpu_ctx = paddle::platform::CPUDeviceContext(); - auto blas = math::GetBlas(cpu_ctx); + auto blas = math::GetBlas( + paddle::platform::CPUDeviceContext()); for (auto j = 0; j < static_cast(ids.size()); ++j) { blas.VSUB(dims1, t_psrever.data() + j * dims1, @@ -798,7 +799,7 @@ void GeoCommunicator::InitSparse() { auto &recv_var_name = ctx.splited_varnames[i]; distributed::VarHandlePtr ret; - ret = rpc_client->AsyncGetVarNoBarrier(endpoints[i], cpu_ctx, local_scope, + ret = rpc_client->AsyncGetVarNoBarrier(ctx.epmap[i], cpu_ctx, local_scope, recv_var_name, recv_var_name); width = recv_t.value().dims()[1]; -- GitLab