提交 69c349a0 编写于 作者: S seiriosPlus

geo sparse init from pserver

上级 c0b84874
...@@ -676,7 +676,8 @@ void GeoCommunicator::RecvSparse(const std::string &varname) { ...@@ -676,7 +676,8 @@ void GeoCommunicator::RecvSparse(const std::string &varname) {
v_delta.resize(numel); v_delta.resize(numel);
auto cpu_ctx = paddle::platform::CPUDeviceContext(); auto cpu_ctx = paddle::platform::CPUDeviceContext();
auto blas = math::GetBlas<platform::CPUDeviceContext, float>(cpu_ctx); auto blas = math::GetBlas<platform::CPUDeviceContext, float>(
paddle::platform::CPUDeviceContext());
for (auto j = 0; j < static_cast<int>(ids.size()); ++j) { for (auto j = 0; j < static_cast<int>(ids.size()); ++j) {
blas.VSUB(dims1, t_psrever.data<float>() + j * dims1, blas.VSUB(dims1, t_psrever.data<float>() + j * dims1,
...@@ -798,7 +799,7 @@ void GeoCommunicator::InitSparse() { ...@@ -798,7 +799,7 @@ void GeoCommunicator::InitSparse() {
auto &recv_var_name = ctx.splited_varnames[i]; auto &recv_var_name = ctx.splited_varnames[i];
distributed::VarHandlePtr ret; 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); recv_var_name, recv_var_name);
width = recv_t.value().dims()[1]; width = recv_t.value().dims()[1];
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册