Function PyMem_SetAllocator 

pub unsafe extern "C" fn PyMem_SetAllocator(
    domain: PyMemAllocatorDomain,
    allocator: *mut PyMemAllocatorEx,
)

Available on crate feature tls and neither PyPy nor GraalPy only.