Method background_buffer_fetch_

mlx/data/stream/Buffered.cpp:23–54

      buffer_(nullptr) {}

std::future<std::shared_ptr<buffer::Buffer>>
Buffered::background_buffer_fetch_() const {
  return pool_->enqueue([this]() -> std::shared_ptr<buffer::Buffer> {
    std::vector<std::future<Sample>> future_buffer;
    {
      std::unique_lock lock(pool_mutex_);
      if (pool_is_alive_) {
        for (int i = 0; i < buffer_size_; i++) {
          future_buffer.push_back(
              pool_->enqueue([this] { return stream_->next(); }));
        }
      }
    }

    std::vector<Sample> buffer;
    for (auto& fsample : future_buffer) {
      Sample sample = fsample.get();
      if (!sample.empty()) {
        buffer.push_back(sample);
      }
    }

    {
      std::unique_lock lock(pool_mutex_);
      if (pool_is_alive_) {
        return on_refill(std::make_shared<FromVector>(buffer));
      }
    }

    return nullptr;
  });
}

Sample Buffered::next() const {
  std::unique_lock lock(mutex_);