mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
89c37be6b7
Pull Request resolved: https://github.com/pytorch/pytorch/pull/136945 Approved by: https://github.com/albanD
23 lines
561 B
C
23 lines
561 B
C
#ifndef THP_EVENT_INC
|
|
#define THP_EVENT_INC
|
|
|
|
#include <c10/core/Event.h>
|
|
#include <torch/csrc/python_headers.h>
|
|
|
|
struct TORCH_API THPEvent {
|
|
PyObject_HEAD
|
|
c10::Event event;
|
|
};
|
|
TORCH_API extern PyTypeObject* THPEventClass;
|
|
TORCH_API extern PyTypeObject THPEventType;
|
|
|
|
TORCH_API void THPEvent_init(PyObject* module);
|
|
TORCH_API PyObject* THPEvent_new(
|
|
c10::DeviceType device_type,
|
|
c10::EventFlag flag);
|
|
inline bool THPEvent_Check(PyObject* obj) {
|
|
return THPEventClass && PyObject_IsInstance(obj, (PyObject*)THPEventClass);
|
|
}
|
|
|
|
#endif // THP_EVENT_INC
|