Skip to content

Commit 7f4b965

Browse files
committed
set status before Finish in prefetch process
1 parent 5aba10b commit 7f4b965

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

paddle/fluid/operators/detail/grpc_server.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,9 +168,9 @@ class RequestPrefetch final : public RequestBase {
168168

169169
SerializeToByteBuffer(out_var_name, outvar, *request_handler_->dev_ctx(),
170170
&reply_);
171+
status_ = FINISH;
171172
responder_.Finish(reply_, ::grpc::Status::OK,
172173
reinterpret_cast<void*>(static_cast<intptr_t>(req_id_)));
173-
status_ = FINISH;
174174
}
175175

176176
protected:

paddle/fluid/operators/listen_and_serv_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,7 @@ void ListenAndServOp::RunImpl(const framework::Scope &scope,
247247
rpc_service_->RegisterRPC(detail::kRequestSend, request_send_handler_.get());
248248
rpc_service_->RegisterRPC(detail::kRequestGet, request_get_handler_.get());
249249
rpc_service_->RegisterRPC(detail::kRequestPrefetch,
250-
request_prefetch_handler_.get(), 1);
250+
request_prefetch_handler_.get());
251251

252252
auto *optimize_block = Attr<framework::BlockDesc *>(kOptimizeBlock);
253253
auto *program = optimize_block->Program();

0 commit comments

Comments
 (0)