torch.cuda.export_dot

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

Return a capture-end hook that dumps the captured graph to path in Graphviz DOT format. Register it with CUDAGraph.register_capture_end_hook(); works for both keep_graph modes since it runs while the template is still live.

Return type:

Callable[[CUDAGraph], None]