CUDAGraph
class torch.cuda.CUDAGraph
Wrapper around a CUDA graph.
Warning:
 This API is in beta and may change in future releases.

capture_begin(pool=None)
  Begins capturing CUDA work on the current stream.

  Typically, you shouldn't call "capture_begin" yourself. Use
  "graph" or "make_graphed_callables()", which call
  "capture_begin" internally.

  Parameters:
     **pool** (*optional*) -- Token (returned by
     "graph_pool_handle()" or "other_Graph_instance.pool()") that
     hints this graph may share memory with the indicated pool.
     See Graph memory management.

capture_end()
  Ends CUDA graph capture on the current stream. After
  "capture_end", "replay" may be called on this instance.

  Typically, you shouldn't call "capture_end" yourself. Use
  "graph" or "make_graphed_callables()", which call "capture_end"
  internally.

debug_dump(debug_path)
  Parameters:
