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).