mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
c54e2c5b41
Pull Request resolved: https://github.com/pytorch/pytorch/pull/164522 Approved by: https://github.com/williamwen42 ghstack dependencies: #164304
24 lines
586 B
C
24 lines
586 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;
|
|
PyObject* weakreflist;
|
|
};
|
|
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
|