"cpu only macro"

revert-3824-remove_grad_op_type
dongzhihong 8 years ago
parent 9a16327b15
commit 69b1b26511

@ -53,6 +53,7 @@ public:
std, std,
seed); seed);
} else { } else {
#ifndef PADDLE_ONLY_CPU
Gaussian( Gaussian(
dynamic_cast<platform::CUDADeviceContext*>(context.device_context_), dynamic_cast<platform::CUDADeviceContext*>(context.device_context_),
output->mutable_data<T>(context.GetPlace()), output->mutable_data<T>(context.GetPlace()),
@ -60,6 +61,7 @@ public:
mean, mean,
std, std,
seed); seed);
#endif
} }
} }
}; };

Loading…
Cancel
Save