From 426c3e1326ea2e127151ad9b16637e30cad58728 Mon Sep 17 00:00:00 2001 From: seiriosPlus Date: Fri, 21 Aug 2020 09:33:14 +0800 Subject: [PATCH] fix parame recv for sparse --- paddle/fluid/operators/distributed/parameter_recv.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/operators/distributed/parameter_recv.cc b/paddle/fluid/operators/distributed/parameter_recv.cc index 376ab6f402d..03b9000999b 100644 --- a/paddle/fluid/operators/distributed/parameter_recv.cc +++ b/paddle/fluid/operators/distributed/parameter_recv.cc @@ -90,7 +90,7 @@ void RecvSelectedRows(const CommContext &rpc_ctx, } else { std::transform(recv_t.rows().begin(), recv_t.rows().end(), std::back_inserter(all_ids), - [&](int64_t id) { return i * pserver_num + id; }); + [&](int64_t id) { return id * pserver_num + i; }); } } -- GitLab