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 7 struct THCPEvent { 8 PyObject_HEAD at::cuda::CUDAEvent cuda_event; 9 }; 10 extern PyObject* THCPEventClass; 11 12 void THCPEvent_init(PyObject* module); 13 THCPEvent_Check(PyObject * obj)14inline bool THCPEvent_Check(PyObject* obj) { 15 return THCPEventClass && PyObject_IsInstance(obj, THCPEventClass); 16 } 17 18 #endif // THCP_EVENT_INC 19