"cpu only macro"

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

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

Loading…
Cancel
Save