xref: /aosp_15_r20/external/pytorch/torch/csrc/cuda/Event.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
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)14 inline bool THCPEvent_Check(PyObject* obj) {
15   return THCPEventClass && PyObject_IsInstance(obj, THCPEventClass);
16 }
17 
18 #endif // THCP_EVENT_INC
19