CUDA 可插拔分配器

类torch.cuda.CUDAPluggableAllocator(path_to_so_file, alloc_fn_name, free_fn_name)[源代码]

CUDA 内存分配器从 so 文件中加载。