We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2b2406e commit 95a2b5fCopy full SHA for 95a2b5f
paddle/fluid/operators/distributed/sendrecvop_utils.cc
@@ -28,10 +28,12 @@ namespace distributed {
28
29
using VarMsg = sendrecv::VariableMessage;
30
31
+#ifdef PADDLE_WITH_CUDA
32
void* GetVarPayLoad(const std::string varname, int64_t size) {
33
platform::CUDAPinnedPlace cuda_pinned;
34
return memory::Alloc(cuda_pinned, size);
35
}
36
+#endif
37
38
void GetTensorPayload(framework::Variable* var,
39
const platform::DeviceContext& ctx, VarMsg* request,
0 commit comments