• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1torch.mtia
2===================================
3
4The MTIA backend is implemented out of the tree, only interfaces are be defined here.
5
6.. automodule:: torch.mtia
7.. currentmodule:: torch.mtia
8
9.. autosummary::
10    :toctree: generated
11    :nosignatures:
12
13    StreamContext
14    current_device
15    current_stream
16    default_stream
17    device_count
18    init
19    is_available
20    is_initialized
21    memory_stats
22    set_device
23    set_stream
24    stream
25    synchronize
26    device
27    set_rng_state
28    get_rng_state
29    DeferredMtiaCallError
30
31Streams and events
32------------------
33.. autosummary::
34    :toctree: generated
35    :nosignatures:
36
37    Event
38    Stream
39