tensorplay documentation — TensorPlay Download

Star–⇄ browse all packagesc

Latest development documentation · Updated 2026-09-08. A documentation snapshot for package 1.0.0.dev20260909 is not available.

On this page

Functions 3

caching_allocator_disabled

functionFull reference ↗

tensorplay.cuda.memory.caching_allocator_disabled()[source]: Context manager that temporarily disables the CUDA caching allocator.

caching_allocator_enable

functionFull reference ↗

tensorplay.cuda.memory.caching_allocator_enable(value: bool = True) → None[source]: Enable or disable the CUDA memory allocator. On by default.

use_mem_pool

functionFull reference ↗

tensorplay.cuda.memory.use_mem_pool(pool)[source]: Route allocations to a MemPool (not supported by this build).