TensorPlay
API symbolscuda
Copy
View MarkdownDownload .md

tensorplay.cuda.graphs.make_graphed_callables

tensorplay.cuda.graphs.make_graphed_callables(callables, sample_args, num_warmup_iters=3, allow_unused_input=False)[source]

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

tensorplay.cuda.graphs.is_current_stream_capturing

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

tensorplay.cuda.host_memory_stats

Return a dictionary of pinned (host) allocator statistics.

Ask DeepWiki