mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
abb313b466
Summary: As title Test Plan: CI tests Reviewed By: joebos Differential Revision: D59036602 Pull Request resolved: https://github.com/pytorch/pytorch/pull/130873 Approved by: https://github.com/hanzlfs
37 lines
642 B
ReStructuredText
37 lines
642 B
ReStructuredText
torch.mtia
|
|
===================================
|
|
|
|
The MTIA backend is implemented out of the tree, only interfaces are be defined here.
|
|
|
|
.. automodule:: torch.mtia
|
|
.. currentmodule:: torch.mtia
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
StreamContext
|
|
current_device
|
|
current_stream
|
|
default_stream
|
|
device_count
|
|
init
|
|
is_available
|
|
is_initialized
|
|
set_device
|
|
set_stream
|
|
stream
|
|
synchronize
|
|
device
|
|
set_rng_state
|
|
get_rng_state
|
|
DeferredMtiaCallError
|
|
|
|
Streams and events
|
|
------------------
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
Event
|
|
Stream
|