From 03f46e3526e43bdffa97f38d33a025504f685679 Mon Sep 17 00:00:00 2001
From: yaoxuefeng <yaoxuefeng@baidu.com>
Date: Thu, 19 Nov 2020 19:33:02 +0800
Subject: [PATCH] fix truncated_gaussian op cuda seed setting (#28678)

---
 paddle/fluid/operators/truncated_gaussian_random_op.cu | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle/fluid/operators/truncated_gaussian_random_op.cu b/paddle/fluid/operators/truncated_gaussian_random_op.cu
index a838c30771a..d4247d9c1d9 100644
--- a/paddle/fluid/operators/truncated_gaussian_random_op.cu
+++ b/paddle/fluid/operators/truncated_gaussian_random_op.cu
@@ -71,7 +71,7 @@ struct TruncatedNormalOffset {
     thrust::minstd_rand rng;
     rng.seed(seed);
     thrust::uniform_real_distribution<T> dist(numeric_min, 1);
-    rng.discard(n);
+    rng.discard(n + offset_);
     T value = dist(rng);
     auto p = a_normal_cdf + (b_normal_cdf - a_normal_cdf) * value;
     return std::sqrt(2.0) * erfinvf(2 * p - 1) * std + mean;
@@ -108,7 +108,7 @@ class GPUTruncatedGaussianRandomKernel : public framework::OpKernel<T> {
           index_sequence_begin, index_sequence_begin + size,
           thrust::device_ptr<T>(data),
           TruncatedNormalOffset<T>(mean, std, std::numeric_limits<T>::min(),
-                                   seed_offset.first, seed_offset.second));
+                                   seed_offset.first, gen_offset));
     }
 
     thrust::transform(
-- 
GitLab