refine buffer receive

7 years ago · 70324911e7
parent 445c74cd93
commit 70324911e7
1 changed files with 1 additions and 1 deletions
--- a/paddle/framework/details/buffered_channel.h
+++ b/paddle/framework/details/buffered_channel.h
@ -71,7 +71,7 @@ bool Buffered<T>::Receive(T* item) {
  std::unique_lock<std::mutex> lock(mu_);
  empty_cond_var_.wait(lock, [this]() { return !channel_.empty() || closed_; });
  bool ret = false;
-  if (!closed_) {
+  if (!channel_.empty()) {
    *item = std::move(channel_.front());
    channel_.pop_front();
    full_cond_var_.notify_one();