Rate this Page

torch.mtia.mtia_graph#

Created On: Nov 10, 2025 | Last Updated On: Nov 10, 2025

The MTIA backend is implemented out of the tree, only interfaces are defined here.

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]#