TensorPlay
API symbolscuda
Copy
View MarkdownDownload .md

tensorplay.cuda.graphs.graph_pool_handle

tensorplay.cuda.graphs.graph_pool_handle()[source]

Return an opaque token representing the id of a graph memory pool.

tensorplay.cuda.graphs.graph

Context-manager that captures CUDA work into a tensorplay.cuda.CUDAGraph .

tensorplay.cuda.graphs.is_current_stream_capturing

Return True if CUDA graph capture is underway on the current thread.

Ask DeepWiki