torch.mtia.mtia_graph

创建于：2025年11月10日 | 最后更新于：2025年12月02日

MTIA 后端是树外实现的，此处仅定义接口。

torch.mtia.mtia_graph.graph_pool_handle()

返回一个代表图内存池 ID 的不透明令牌。

返回类型:

tuple[int, int]

class torch.mtia.mtia_graph.MTIAGraph(keep_graph=False)

MTIA 图的包装器。

返回类型:

自我

capture_begin(pool)

开始捕获 MTIA 图。

capture_end()

结束 MTIA 图的捕获。

instantiate()

实例化捕获的 MTIA 图。

pool()

返回一个代表此图内存池 ID 的不透明令牌。

返回类型:

tuple[int, int]

replay()

重放捕获的 MTIA 图。

reset()

销毁捕获的图并重置状态。

class torch.mtia.mtia_graph.graph(mtia_graph, pool=None, stream=None)