1 #ifndef THCP_STREAM_INC 2 #define THCP_STREAM_INC 3 4 #include <c10/cuda/CUDAStream.h> 5 #include <torch/csrc/Stream.h> 6 #include <torch/csrc/python_headers.h> 7 8 // NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init) 9 struct THCPStream : THPStream { 10 at::cuda::CUDAStream cuda_stream; 11 }; 12 extern PyObject* THCPStreamClass; 13 14 void THCPStream_init(PyObject* module); 15 THCPStream_Check(PyObject * obj)16inline bool THCPStream_Check(PyObject* obj) { 17 return THCPStreamClass && PyObject_IsInstance(obj, THCPStreamClass); 18 } 19 20 #endif // THCP_STREAM_INC 21