blob: 1b654415bb7f938072aad6a53d48745c36e50b7a [file] [log] [blame]
#ifndef THCP_EVENT_INC
#define THCP_EVENT_INC
#include <ATen/cuda/CUDAEvent.h>
#include <torch/csrc/python_headers.h>
// NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init)
struct THCPEvent {
PyObject_HEAD at::cuda::CUDAEvent cuda_event;
};
extern PyObject* THCPEventClass;
void THCPEvent_init(PyObject* module);
inline bool THCPEvent_Check(PyObject* obj) {
return THCPEventClass && PyObject_IsInstance(obj, THCPEventClass);
}
#endif // THCP_EVENT_INC