Lines Matching full:xpu

5 #include <c10/xpu/XPUFunctions.h>
7 namespace c10::xpu {
34  * a SYCL queue, which allows asynchronous execution of XPU tasks.
41   /// will raise an error if the Stream is not, in fact, a XPU stream.
43     TORCH_CHECK(stream_.device_type() == DeviceType::XPU);  in XPUStream()
63   /// XPU stream).
68   /// Get the XPU device type that this stream is associated with.
70     return DeviceType::XPU;  in device_type()
73   /// Get the XPU device index that this stream is associated with.
79   /// guaranteed to be a XPU device.
81     return Device(DeviceType::XPU, device_index());  in device()
161  * Get the current XPU stream, for the passed XPU device, or for the current
180 } // namespace c10::xpu
184 struct hash<c10::xpu::XPUStream> {
185   size_t operator()(c10::xpu::XPUStream s) const noexcept {