1 files changed, 0 insertions, 43 deletions
diff --git a/torch/lib/c10d/CUDAUtils.hpp b/torch/lib/c10d/CUDAUtils.hpp
index c0db85e7ed..b8e99e32ca 100644
--- a/torch/lib/c10d/CUDAUtils.hpp
+++ b/torch/lib/c10d/CUDAUtils.hpp
@@ -1,7 +1,5 @@
 #pragma once
 
-typedef struct CUDAStreamInternals THCStream;
-
 #include <algorithm>
 
 #include <cuda.h>
@@ -50,45 +48,4 @@ class CUDAEvent {
   cudaEvent_t event_;
 };
 
-// RAII wrapper for CUDA streams.
-//
-// This wrapper uses THCStream instead of cudaStream_t because we need
-// to interact with the THC API for selecting the current stream.
-// Doing this without having a THCStream pointer is cumbersome.
-//
-class CUDAStream {
- public:
-  CUDAStream(THCStream* stream) : stream_(stream) {}
-
-  CUDAStream() : CUDAStream(nullptr) {}
-
-  ~CUDAStream();
-
-  static CUDAStream create();
-
-  // Must not be copyable.
-  CUDAStream& operator=(const CUDAStream&) = delete;
-  CUDAStream(const CUDAStream&) = delete;
-
-  // Must be move constructable.
-  CUDAStream(CUDAStream&& other) {
-    std::swap(stream_, other.stream_);
-  }
-
-  // Must be move assignable.
-  CUDAStream& operator=(CUDAStream&& other) {
-    std::swap(stream_, other.stream_);
-    return *this;
-  }
-
-  cudaStream_t getStream() const;
-
-  THCStream* getTHCStream() {
-    return stream_;
-  }
-
- protected:
-  THCStream* stream_;
-};
-
 } // namespace c10d