CUDAGraph

class tensorplay.cuda.graphs.CUDAGraph[source]

Wrapper around a CUDA graph, mirroring torch.cuda.CUDAGraph.

instantiate()[source]

Compile the captured template (happens automatically at replay).

reset()[source]

Destroy the executable and free its private memory pool.

All tensors allocated during the capture must be released first.