Follow comments

paddle/fluid/operators/reader/blocking_queue.h

@@ -25,6 +25,11 @@ namespace reader {
 
 template <typename T>
 class BlockingQueue {
+  // BlockingQueue is for buffered reading and is supposed to use only the
+  // reader package. It is true that we could and we should have been using
+  // framework::Channel, but which has currently a deadlock bug. BlockingQueue
+  // is a workaround and a simplified version of framework::Channel as it
+  // doesn't support GPU and it implements on buffered blocking queue.
  public:
   explicit BlockingQueue(size_t capacity)
       : capacity_(capacity), closed_(false) {
@@ -37,27 +42,29 @@ class BlockingQueue {
     std::unique_lock<std::mutex> lock(mutex_);
     send_cv_.wait(lock, [&] { return queue_.size() < capacity_ || closed_; });
     if (closed_) {
+      VLOG(5)
+          << "WARNING: Sending an element to a closed reader::BlokcingQueue.";
       return false;
-    } else {
+    }
     PADDLE_ENFORCE_LT(queue_.size(), capacity_);
     queue_.push_back(elem);
     receive_cv_.notify_one();
     return true;
   }
-  }
 
   bool Send(T&& elem) {
     std::unique_lock<std::mutex> lock(mutex_);
     send_cv_.wait(lock, [&] { return queue_.size() < capacity_ || closed_; });
     if (closed_) {
+      VLOG(5)
+          << "WARNING: Sending an element to a closed reader::BlokcingQueue.";
       return false;
-    } else {
+    }
     PADDLE_ENFORCE_LT(queue_.size(), capacity_);
     queue_.emplace_back(std::move(elem));
     receive_cv_.notify_one();
     return true;
   }
-  }
 
   bool Receive(T* elem) {
     std::unique_lock<std::mutex> lock(mutex_);
@@ -86,16 +93,6 @@ class BlockingQueue {
     return closed_;
   }
 
-  bool CanSend() {
-    std::lock_guard<std::mutex> lock(mutex_);
-    return !closed_ && queue_.size() < capacity_;
-  }
-
-  bool CanReceive() {
-    std::lock_guard<std::mutex> lock(mutex_);
-    return !queue_.empty();
-  }
-
   size_t Cap() {
     std::lock_guard<std::mutex> lock(mutex_);
     return capacity_;

paddle/fluid/operators/reader/create_double_buffer_reader_op.cc

@@ -55,8 +55,6 @@ class DoubleBufferReader : public framework::DecoratedReader {
   ~DoubleBufferReader() { EndPrefetcher(); }
 
  private:
-  bool HasNext() const;
-
   void StartPrefetcher() {
     channel_ = new reader::BlockingQueue<size_t>(kChannelSize);
     prefetcher_ = std::thread([this] { PrefetchThreadFunc(); });
@@ -139,17 +137,16 @@ class CreateDoubleBufferReaderOpMaker : public DecoratedReaderMakerBase {
 };
 
 void DoubleBufferReader::ReadNext(std::vector<framework::LoDTensor>* out) {
-  out->clear();
-  if (HasNext()) {
   size_t cached_tensor_id;
-    channel_->Receive(&cached_tensor_id);
+  if (channel_->Receive(&cached_tensor_id)) {
     if (platform::is_gpu_place(place_)) {
       *out = gpu_tensor_cache_[cached_tensor_id];
-      ctxs_[cached_tensor_id]->Wait();
     } else {
       // CPU place
       *out = cpu_tensor_cache_[cached_tensor_id];
     }
+  } else {
+    out->clear();
   }
 }
 
@@ -159,12 +156,6 @@ void DoubleBufferReader::ReInit() {
   StartPrefetcher();
 }
 
-bool DoubleBufferReader::HasNext() const {
-  while (!channel_->IsClosed() && !channel_->CanReceive()) {
-  }
-  return channel_->CanReceive();
-}
-
 void DoubleBufferReader::PrefetchThreadFunc() {
   VLOG(5) << "A new prefetch thread starts.";
   size_t cached_tensor_id = 0;

paddle/fluid/operators/reader/open_files_op.cc

@@ -37,7 +37,6 @@ class MultiFileReader : public framework::ReaderBase {
   ~MultiFileReader() { EndScheduler(); }
 
  private:
-  bool HasNext();
   void StartNewScheduler();
   void EndScheduler();
   void ScheduleThreadFunc();
@@ -54,9 +53,8 @@ class MultiFileReader : public framework::ReaderBase {
 };
 
 void MultiFileReader::ReadNext(std::vector<framework::LoDTensor>* out) {
+  if (!buffer_->Receive(out)) {
     out->clear();
-  if (HasNext()) {
-    buffer_->Receive(out);
   }
 }
 
@@ -65,12 +63,6 @@ void MultiFileReader::ReInit() {
   StartNewScheduler();
 }
 
-bool MultiFileReader::HasNext() {
-  while (!buffer_->IsClosed() && !buffer_->CanReceive()) {
-  }
-  return buffer_->CanReceive();
-}
-
 void MultiFileReader::StartNewScheduler() {
   size_t thread_num = prefetchers_.size();
   waiting_file_idx_ = new reader::BlockingQueue<size_t>(file_names_.size());