1 #pragma once 2 3 #include <c10/xpu/XPUStream.h> 4 #include <torch/csrc/Stream.h> 5 #include <torch/csrc/python_headers.h> 6 7 // NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init) 8 struct THXPStream : THPStream { 9 at::xpu::XPUStream xpu_stream; 10 }; 11 extern PyObject* THXPStreamClass; 12 13 void THXPStream_init(PyObject* module); 14 THXPStream_Check(PyObject * obj)15inline bool THXPStream_Check(PyObject* obj) { 16 return THXPStreamClass && PyObject_IsInstance(obj, THXPStreamClass); 17 } 18