fix gpu build error

revert-3824-remove_grad_op_type
qijun 8 years ago
parent 1918ad8759
commit aff90d8ee7

@ -30,7 +30,7 @@ struct GaussianGenerator {
__host__ __device__ T operator()(const unsigned int n) const {
thrust::minstd_rand rng;
rng.seed(seed_);
thrust::normal_distribution<T> dist(min_, max_);
thrust::normal_distribution<T> dist(mean_, std_);
rng.discard(n);
return dist(rng);
}
@ -62,4 +62,4 @@ class GPUGaussianRandomKernel : public framework::OpKernel {
} // namespace paddle
REGISTER_OP_GPU_KERNEL(gaussian_random,
paddle::operators::GPUGaussianRandomKernel<float>);
paddle::operators::GPUGaussianRandomKernel<float>);

Loading…
Cancel
Save