torch.cuda.export_dot

torch.cuda.export_dot(path, *, verbose=True) [source]

Возвращает хук завершения захвата, который сохраняет захваченный граф в path в формате Graphviz DOT. Зарегистрируйте его с помощью CUDAGraph.register_capture_end_hook(); работает в обоих режимах keep_graph, поскольку выполняется, пока шаблон ещё активен.

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

Callable[[CUDAGraph], None]