1#ifndef THCP_EVENT_INC
2#define THCP_EVENT_INC
3
4#include <ATen/cuda/CUDAEvent.h>
5#include <torch/csrc/python_headers.h>
6
7struct THCPEvent {
8 PyObject_HEAD at::cuda::CUDAEvent cuda_event;
9};
10extern PyObject* THCPEventClass;
11
12void THCPEvent_init(PyObject* module);
13
14inline bool THCPEvent_Check(PyObject* obj) {
15 return THCPEventClass && PyObject_IsInstance(obj, THCPEventClass);
16}
17
18#endif // THCP_EVENT_INC
19