fix mac build of sendrecvop_utils

7 years ago · 938390b38d
parent c5c17a14f8
commit 938390b38d
1 changed files with 2 additions and 0 deletions
--- a/paddle/fluid/operators/distributed/sendrecvop_utils.cc
+++ b/paddle/fluid/operators/distributed/sendrecvop_utils.cc
@ -28,10 +28,12 @@ namespace distributed {

 using VarMsg = sendrecv::VariableMessage;

+#ifdef PADDLE_WITH_CUDA
 void* GetVarPayLoad(const std::string varname, int64_t size) {
  platform::CUDAPinnedPlace cuda_pinned;
  return memory::Alloc(cuda_pinned, size);
 }
+#endif

 void GetTensorPayload(framework::Variable* var,
                      const platform::DeviceContext& ctx, VarMsg* request,