StreamContext
Context-manager that selects a given stream.
current_device
Return the index of a currently selected device.
current_stream
Return the currently selected Stream for a given device.
device
Context-manager that changes the selected device.
device_count
Return the number of XPU device available.
device_of
Context-manager that changes the current device to that of given object.
get_arch_list
Return list XPU architectures this library was compiled for.
get_device_capability
Get the xpu capability of a device.
get_device_name
Get the name of a device.
get_device_properties
Get the properties of a device.
get_gencode_flags
Return XPU AOT(ahead-of-time) build flags this library was compiled with.
get_stream_from_external
Return a Stream from an external SYCL queue.
init
Initialize PyTorch's XPU state.
is_available
Return a bool indicating if XPU is currently available.
is_initialized
Return whether PyTorch's XPU state has been initialized.
set_device
Set the current device.
set_stream
Set the current stream.This is a wrapper API to set the stream.
stream
Wrap around the Context-manager StreamContext that selects a given stream.
synchronize
Wait for all kernels in all streams on a XPU device to complete.
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4