Remove unnecessary header files (#49)

tanquer · meta-codesync[bot] · commit 78e955902397 · 2025-11-17T13:02:02.000-08:00
Summary: Pull Request resolved: #49 Remove unnecessary header file to make gloo backend not directly depend on `cuda_runtime.h`. Reviewed By: fduwjj Differential Revision: D86274376 fbshipit-source-id: 105a6707caeb0bc03f6b766f7294f5391dc84d80
diff --git a/comms/torchcomms/gloo/TorchCommGloo.hpp b/comms/torchcomms/gloo/TorchCommGloo.hpp
@@ -3,14 +3,8 @@
 #pragma once
 
 #include <atomic>
-#include <chrono>
-#include <condition_variable>
 #include <memory>
-#include <mutex>
-#include <queue>
 #include <string_view>
-#include <thread>
-#include <unordered_map>
 #include <vector>
 
 #include <ATen/ATen.h>
diff --git a/comms/torchcomms/gloo/TorchWorkGloo.hpp b/comms/torchcomms/gloo/TorchWorkGloo.hpp
@@ -2,17 +2,7 @@
 
 #pragma once
 
-#include <atomic>
-#include <chrono>
-#include <memory>
-#include <mutex>
-#include <optional>
-#include <queue>
-#include <unordered_map>
-
 #include <ATen/ATen.h>
-#include <cuda_runtime.h> // @manual=third-party//cuda:cuda-lazy
-#include <vector>
 #include "comms/torchcomms/TorchCommTracing.hpp"
 #include "comms/torchcomms/TorchWork.hpp"
 
