| torch.xpu |
| =================================== |
| .. automodule:: torch.xpu |
| .. currentmodule:: torch.xpu |
| |
| .. autosummary:: |
| :toctree: generated |
| :nosignatures: |
| |
| StreamContext |
| current_device |
| current_stream |
| device |
| device_count |
| device_of |
| empty_cache |
| get_device_capability |
| get_device_name |
| get_device_properties |
| init |
| is_available |
| is_initialized |
| set_device |
| set_stream |
| stream |
| synchronize |
| |
| Streams and events |
| ------------------ |
| .. autosummary:: |
| :toctree: generated |
| :nosignatures: |
| |
| Event |
| Stream |
| |
| |
| .. This module needs to be documented. Adding here in the meantime |
| .. for tracking purposes |
| .. py:module:: torch.xpu.streams |
