TensorPlay
API symbolscuda
Copy
View MarkdownDownload .md

tensorplay.cuda.graphs.is_current_stream_capturing

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

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

tensorplay.cuda.graphs.graph_pool_handle

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

tensorplay.cuda.graphs.make_graphed_callables

Callables that run per-iteration with CUDA graph capture (not supported).

Ask DeepWiki