graph

class torch.xpu.graph(xpu_graph, pool=None, stream=None) [source]

Менеджер контекста, который захватывает работу XPU в объект torch.xpu.XPUGraph для последующего воспроизведения.

Параметры:

• xpu_graph (torch.xpu.XPUGraph) – Объект графа, используемый для захвата.
• pool (необязательно) – Непрозрачный токен (возвращаемый вызовом graph_pool_handle() или other_Graph_instance.pool()), указывающий, что при захвате этого графа может использоваться память из указанного пула.
• stream (torch.xpu.Stream, необязательно) – Если указан, будет установлен как текущий поток в контексте. Если не указан, graph устанавливает собственный внутренний вспомогательный поток как текущий поток в контексте.

Примечание

Для эффективного совместного использования памяти, если вы передаёте pool, использованный при предыдущем захвате, и при предыдущем захвате был указан явный аргумент stream, следует передать тот же аргумент stream и для этого захвата.