diff --git a/paddle/fluid/memory/memcpy.cc b/paddle/fluid/memory/memcpy.cc index c76d448013f04f0053c6245b6660939fb100f9f7..1a0a77bcc4c8e47cebcf09e4607d012b4944c756 100644 --- a/paddle/fluid/memory/memcpy.cc +++ b/paddle/fluid/memory/memcpy.cc @@ -245,7 +245,7 @@ void Copy(platform::CPUPlace dst_place, platform::NPUMemcpyAsync(dst, src, num, ACL_MEMCPY_DEVICE_TO_HOST, stream); } else { platform::DeviceContextPool& pool = platform::DeviceContextPool::Instance(); - static_cast(pool.Get(dst_place))->Wait(); + static_cast(pool.Get(src_place))->Wait(); platform::RecordEvent record_event("GpuMemcpySync:NPU->CPU"); platform::NPUMemcpySync(dst, src, num, ACL_MEMCPY_DEVICE_TO_HOST);