1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
| #ifndef THCP_EVENT_INC
| #define THCP_EVENT_INC
|
| #include <ATen/cuda/CUDAEvent.h>
| #include <torch/csrc/python_headers.h>
| #include <THC/THC.h>
|
| 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
|
|
