mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
eac02f85cf
Summary: In my last PR I've missed CUDA and distributed folders, fixing this now This change is autogenerated by `python tool/clang_tidy.py -s` Pull Request resolved: https://github.com/pytorch/pytorch/pull/57235 Reviewed By: janeyx99 Differential Revision: D28084444 Pulled By: malfet fbshipit-source-id: bf222f69ee90c7872c3cb0931e8cdb84f0cb3cda
22 lines
524 B
C
22 lines
524 B
C
#ifndef THCP_STREAM_INC
|
|
#define THCP_STREAM_INC
|
|
|
|
#include <torch/csrc/Stream.h>
|
|
#include <c10/cuda/CUDAStream.h>
|
|
#include <torch/csrc/python_headers.h>
|
|
#include <THC/THC.h>
|
|
|
|
// NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init)
|
|
struct THCPStream : THPStream{
|
|
at::cuda::CUDAStream cuda_stream;
|
|
};
|
|
extern PyObject *THCPStreamClass;
|
|
|
|
void THCPStream_init(PyObject *module);
|
|
|
|
inline bool THCPStream_Check(PyObject* obj) {
|
|
return THCPStreamClass && PyObject_IsInstance(obj, THCPStreamClass);
|
|
}
|
|
|
|
#endif // THCP_STREAM_INC
|