fix parame recv for sparse

paddle/fluid/operators/distributed/communicator.cc

@@ -628,9 +628,7 @@ void GeoCommunicator::RecvByCommunicator() {
       if (recv_ctx.is_sparse) {
         RecvSparse(var_name);
       } else {
-        VLOG(1) << "recv dense " << var_name << " begin";
         RecvDense(var_name);
-        VLOG(1) << "recv dense " << var_name << " done";
       }
     };
     tasks.emplace_back(send_threadpool_->enqueue(std::move(recv_task)));

paddle/fluid/operators/distributed/parameter_recv.cc

@@ -84,9 +84,14 @@ void RecvSelectedRows(const CommContext &rpc_ctx,
     ids_num += recv_t.rows().size();
     width = recv_t.value().dims()[1];
 
+    if (rpc_ctx.is_distributed) {
+      std::copy(recv_t.rows().begin(), recv_t.rows().end(),
+                std::back_inserter(all_ids));
+    } else {
       std::transform(recv_t.rows().begin(), recv_t.rows().end(),
                      std::back_inserter(all_ids),
-                   [&](int64_t id) { return id * pserver_num + i; });
+                     [&](int64_t id) { return i * pserver_num + id; });
+    }
   }
 
   auto *var = scope.FindVar(rpc_ctx.var_name);