Skip to content

Commit 5aba10b

Browse files
committed
set the thread pool of prefetch to 1 to fix a bug
1 parent 8fb78f6 commit 5aba10b

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

paddle/fluid/operators/detail/grpc_server.cc

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,6 @@ class RequestPrefetch final : public RequestBase {
158158
std::string in_var_name = request_->Varname();
159159
std::string out_var_name = request_->OutVarname();
160160
VLOG(3) << "in_var_name: " << in_var_name
161-
<< "out_var_name: " << out_var_name
162161
<< " RequestPrefetch: " << out_var_name;
163162

164163
auto scope = request_->GetMutableLocalScope();

paddle/fluid/operators/listen_and_serv_op.cc

Lines changed: 3 additions & 2 deletions
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());
250+
request_prefetch_handler_.get(), 1);
251251

252252
auto *optimize_block = Attr<framework::BlockDesc *>(kOptimizeBlock);
253253
auto *program = optimize_block->Program();
@@ -263,7 +263,8 @@ void ListenAndServOp::RunImpl(const framework::Scope &scope,
263263
prefetch_var_name_to_block_id_str) {
264264
std::vector<std::string> pieces;
265265
split(prefetch_var_name_and_id, ':', &pieces);
266-
VLOG(3) << "after split, grad = " << pieces[0] << ", id=" << pieces[1];
266+
VLOG(3) << "after split, prefetch_var = " << pieces[0]
267+
<< ", id=" << pieces[1];
267268
PADDLE_ENFORCE_EQ(pieces.size(), 2);
268269

269270
int block_id = std::stoi(pieces[1]);

0 commit comments

Comments
 (0)