1 #pragma once 2 3 #include <ATen/xpu/XPUEvent.h> 4 #include <torch/csrc/python_headers.h> 5 6 struct THXPEvent { 7 PyObject_HEAD at::xpu::XPUEvent xpu_event; 8 }; 9 extern PyObject* THXPEventClass; 10 11 void THXPEvent_init(PyObject* module); 12 THXPEvent_Check(PyObject * obj)13inline bool THXPEvent_Check(PyObject* obj) { 14 return THXPEventClass && PyObject_IsInstance(obj, THXPEventClass); 15 } 16