torch.mtia.mtia_graph
Created On: Nov 10, 2025 | Last Updated On: Dec 02, 2025
The MTIA backend is implemented out of the tree, only interfaces are defined here.
-
torch.mtia.mtia_graph.graph_pool_handle()[source]
Return an opaque token representing the id of a graph memory pool.
- Return type:
tuple[int, int]
-
class torch.mtia.mtia_graph.MTIAGraph(keep_graph=False)[source]
Wrapper around a MTIA graph.
- Return type:
Self
-
capture_begin(pool)[source]
Begin capturing a MTIA graph.
-
capture_end()[source]
End the capture of a MTIA graph.
-
instantiate()[source]
Instantiate the captured MTIA graph.
-
pool()[source]
Return an opaque token representing the id of this graph’s memory pool
- Return type:
tuple[int, int]
-
replay()[source]
Replay the captured MTIA graph.
-
reset()[source]
Destroy the captured graph and reset the states.
-
class torch.mtia.mtia_graph.graph(mtia_graph, pool=None, stream=None)[source]
