Method FreeImpl

paddle/phi/core/memory/allocation/cuda_managed_allocator.cc:37–46

bool CUDAManagedAllocator::IsAllocThreadSafe() const { return true; }

void CUDAManagedAllocator::FreeImpl(phi::Allocation* allocation) {
  PADDLE_ENFORCE_EQ(
      allocation->place(),
      place_,
      common::errors::PermissionDenied(
          "GPU memory is freed in incorrect device. This may be a bug"));
  platform::RecordedGpuFree(
      allocation->ptr(), allocation->size(), place_.device);
  delete allocation;
}

phi::Allocation* CUDAManagedAllocator::AllocateImpl(size_t size) {
  std::call_once(once_flag_, [this] { platform::SetDeviceId(place_.device); });