torch.mtia.mtia_graph

Создано: 10 нояб. 2025 | Последнее обновление: 02 дек. 2025

Бэкенд MTIA реализован вне дерева исходного кода, здесь определены только интерфейсы.

torch.mtia.mtia_graph.graph_pool_handle() [исходный код]

Возвращает непрозрачный токен, представляющий идентификатор пула памяти графа.

Тип возвращаемого значения:

tuple[int, int]

class torch.mtia.mtia_graph.MTIAGraph(keep_graph=False) [исходный код]

Обёртка для графа MTIA.

Тип возвращаемого значения:

Self

capture_begin(pool) [исходный код]

Начинает захват графа MTIA.

capture_end() [исходный код]

Завершает захват графа MTIA.

instantiate() [исходный код]

Создаёт экземпляр захваченного графа MTIA.

pool() [исходный код]

Возвращает непрозрачный токен, представляющий идентификатор пула памяти этого графа

Тип возвращаемого значения:

tuple[int, int]

replay() [исходный код]

Повторно запускает захваченный граф MTIA.

reset() [исходный код]

Уничтожает захваченный граф и сбрасывает состояния.

class torch.mtia.mtia_graph.graph(mtia_graph, pool=None, stream=None) [исходный код]