Hello everyone. What is the current/recommended  API to use for custom extensions that need to perform allocations and  want  to use/share the memory  allocation pool created by the PyTorch library?  For instance functions/operators that uses the Thrust library uses a series of wrappers that finally calls c10::cuda::CUDACachingAllocator::raw_alloc and c10::cuda::CUDACachingAllocator::raw_delete. These two functions should be ok to be called from a external code/library?
thanks