[custom runtime] clear headers (#40845)

paddle/phi/backends/callback_manager.cc

@@ -16,16 +16,18 @@
 #include "paddle/fluid/platform/device/device_wrapper.h"
 #include "paddle/fluid/platform/enforce.h"
 
+#include <ThreadPool.h>
+
 namespace phi {
 
 CallbackManager::CallbackManager(stream::Stream *stream)
-    : stream_(stream), thread_pool_(1) {}
+    : 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] {
+    last_future_ = thread_pool_->enqueue([callback_func] {
       std::unique_ptr<std::function<void()>> releaser(callback_func);
       (*callback_func)();
     });

paddle/phi/backends/callback_manager.h

@@ -14,8 +14,6 @@
 
 #pragma once
 
-#include <ThreadPool.h>
-
 #ifdef PADDLE_WITH_CUDA
 #include <cuda.h>
 #include <cuda_runtime.h>
@@ -30,6 +28,8 @@
 #include <memory>
 #include <mutex>  // NOLINT
 
+class ThreadPool;
+
 namespace phi {
 
 namespace stream {
@@ -50,7 +50,7 @@ class CallbackManager {
 
  private:
   stream::Stream* stream_;
-  mutable ::ThreadPool thread_pool_;
+  mutable std::shared_ptr<::ThreadPool> thread_pool_;
   mutable std::mutex mtx_;
   mutable std::future<void> last_future_;
 };

paddle/phi/backends/device_base.h

@@ -14,6 +14,8 @@
 
 #pragma once
 #ifdef PADDLE_WITH_CUSTOM_DEVICE
+#include <vector>
+
 #include "paddle/phi/backends/event.h"
 #include "paddle/phi/backends/stream.h"