mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
1764f3a9c8
Fix several gramma errors in PyTorch docs. Pull Request resolved: https://github.com/pytorch/pytorch/pull/166158 Approved by: https://github.com/yewentao256, https://github.com/cyyever, https://github.com/ezyang
53 lines
817 B
Markdown
53 lines
817 B
Markdown
# torch.mtia
|
|
|
|
The MTIA backend is implemented out of the tree, only interfaces are defined here.
|
|
|
|
```{eval-rst}
|
|
.. automodule:: torch.mtia
|
|
```
|
|
|
|
```{eval-rst}
|
|
.. currentmodule:: torch.mtia
|
|
```
|
|
|
|
```{eval-rst}
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
StreamContext
|
|
current_device
|
|
current_stream
|
|
default_stream
|
|
device_count
|
|
init
|
|
is_available
|
|
is_bf16_supported
|
|
is_initialized
|
|
memory_stats
|
|
get_device_capability
|
|
empty_cache
|
|
record_memory_history
|
|
snapshot
|
|
attach_out_of_memory_observer
|
|
set_device
|
|
set_stream
|
|
stream
|
|
synchronize
|
|
device
|
|
set_rng_state
|
|
get_rng_state
|
|
DeferredMtiaCallError
|
|
```
|
|
|
|
## Streams and events
|
|
|
|
```{eval-rst}
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
Event
|
|
Stream
|
|
```
|