From aff90d8ee78be398b2984d63f2eb985f15f430d1 Mon Sep 17 00:00:00 2001 From: qijun Date: Tue, 22 Aug 2017 04:34:35 +0000 Subject: [PATCH] fix gpu build error --- paddle/operators/gaussian_random_op.cu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/operators/gaussian_random_op.cu b/paddle/operators/gaussian_random_op.cu index 1d312e7b5..018a4bfcb 100644 --- a/paddle/operators/gaussian_random_op.cu +++ b/paddle/operators/gaussian_random_op.cu @@ -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 dist(min_, max_); + thrust::normal_distribution 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); \ No newline at end of file + paddle::operators::GPUGaussianRandomKernel); -- GitLab