mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 13:44:15 +08:00
1aa9099839
# Motivation refer to [#118504](https://github.com/pytorch/pytorch/pull/118504), enabling clang-tidy in `torch/csrc/xpu`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/120616 Approved by: https://github.com/albanD
18 lines
437 B
C
18 lines
437 B
C
#pragma once
|
|
|
|
#include <c10/xpu/XPUStream.h>
|
|
#include <torch/csrc/Stream.h>
|
|
#include <torch/csrc/python_headers.h>
|
|
|
|
// NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init)
|
|
struct THXPStream : THPStream {
|
|
at::xpu::XPUStream xpu_stream;
|
|
};
|
|
extern PyObject* THXPStreamClass;
|
|
|
|
void THXPStream_init(PyObject* module);
|
|
|
|
inline bool THXPStream_Check(PyObject* obj) {
|
|
return THXPStreamClass && PyObject_IsInstance(obj, THXPStreamClass);
|
|
}
|