API symbolscudaCopyCopy as MarkdownCopy page linkView MarkdownDownload .mdOpenAIOpen in ChatGPTClaudeOpen in ClaudePerplexityOpen in PerplexityGrokOpen in Grok CUDAPluggableAllocator class tensorplay.cuda.CUDAPluggableAllocator(so_file: str, alloc_fn_name: str = 'my_alloc', free_fn_name: str = 'my_free')[source] CUDA memory allocator plugin (not supported by this TensorPlay build). Help improve this pageFound an error, an unclear step, or a missing example?Was this page helpful?YesNoSuggest an editReport an issueCUDAGraphWrapper around a CUDA graph, mirroring torch.cuda.CUDAGraph .deviceContext-manager that changes the selected device.
