fix mac build of sendrecvop_utils (#12272)

7 years ago · 95a2b5f56a
parent 2b2406e58b
commit 95a2b5f56a
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,