From c770d5c0ba55f33ec1f34db5d70e11cee7f98c2b Mon Sep 17 00:00:00 2001 From: Yancey Date: Fri, 25 May 2018 16:27:40 +0800 Subject: [PATCH] fix build error with testing and gpu on (#10932) --- paddle/fluid/operators/detail/sendrecvop_utils.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/paddle/fluid/operators/detail/sendrecvop_utils.cc b/paddle/fluid/operators/detail/sendrecvop_utils.cc index 3bae56532d6..507b4654356 100644 --- a/paddle/fluid/operators/detail/sendrecvop_utils.cc +++ b/paddle/fluid/operators/detail/sendrecvop_utils.cc @@ -149,12 +149,14 @@ void SerializeToByteBuffer(const std::string& name, framework::Variable* var, } if (platform::is_gpu_place(ctx.GetPlace())) { +#ifdef PADDLE_WITH_CUDA // GPU data is copied to CPU buffer when sending, // free the buffer when possible. destroy_callback = [](void* backing) { platform::CUDAPinnedPlace cuda_pinned; memory::Free(cuda_pinned, backing); }; +#endif } std::string header; -- GitLab