File tree Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -151,6 +151,7 @@ bool VariableResponse::CopySelectRowsData(
151
151
::google::protobuf::io::CodedInputStream* input,
152
152
const platform::DeviceContext& ctx, int length) {
153
153
auto * slr = GetVar ()->GetMutable <framework::SelectedRows>();
154
+ slr->mutable_rows ()->clear ();
154
155
slr->mutable_rows ()->resize (length /
155
156
framework::SizeOfType (typeid (int64_t ))); // int64
156
157
int64_t * rows_data = slr->mutable_rows ()->data ();
Original file line number Diff line number Diff line change @@ -165,12 +165,13 @@ void ListenAndServOp::RunSyncLoop(
165
165
recv_scope);
166
166
VLOG (2 ) << " run all blocks spent " << GetTimestamp () - ts << " (ms)" ;
167
167
168
- rpc_service_->SetCond (distributed::kRequestGet );
169
- rpc_service_->WaitBarrier (distributed::kRequestGet );
170
- rpc_service_->ResetBarrierCounter ();
171
168
// reset received sparse vars to avoid reuse it in the next mini-batch
172
169
dynamic_cast <distributed::RequestSendHandler *>(request_send_handler_.get ())
173
170
->ResetSparseVarRecorder ();
171
+
172
+ rpc_service_->SetCond (distributed::kRequestGet );
173
+ rpc_service_->WaitBarrier (distributed::kRequestGet );
174
+ rpc_service_->ResetBarrierCounter ();
174
175
} // while(true)
175
176
}
176
177
You can’t perform that action at this time.
0 commit comments