Method AddCallback

paddle/phi/backends/callback_manager.cc:27–39

    : stream_(stream), thread_pool_(new ::ThreadPool(1)) {}

void CallbackManager::AddCallback(std::function<void()> callback) const {
  auto *callback_func = new std::function<void()>(std::move(callback));
  auto *func = new std::function<void()>([this, callback_func] {
    std::lock_guard<std::mutex> lock(mtx_);
    last_future_ = thread_pool_->enqueue([callback_func] {
      std::unique_ptr<std::function<void()>> releaser(callback_func);
      (*callback_func)();
    });
  });
  phi::DeviceGuard guard(stream_->GetPlace());
  DeviceManager::GetDeviceWithPlace(stream_->GetPlace())
      ->AddCallback(stream_, func);
}

void CallbackManager::Wait() const {
  phi::DeviceGuard guard(stream_->GetPlace());