fix truncated_gaussian seed (#28777)

musl/fix_failed_unittests_in_musl
yaoxuefeng 5 years ago committed by GitHub
parent de528981e5
commit 71c1cd1408
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -109,12 +109,12 @@ class GPUTruncatedGaussianRandomKernel : public framework::OpKernel<T> {
thrust::device_ptr<T>(data),
TruncatedNormalOffset<T>(mean, std, std::numeric_limits<T>::min(),
seed_offset.first, gen_offset));
} else {
thrust::transform(
index_sequence_begin, index_sequence_begin + size,
thrust::device_ptr<T>(data),
TruncatedNormal<T>(mean, std, std::numeric_limits<T>::min(), seed));
}
thrust::transform(
index_sequence_begin, index_sequence_begin + size,
thrust::device_ptr<T>(data),
TruncatedNormal<T>(mean, std, std::numeric_limits<T>::min(), seed));
}
};

Loading…
Cancel
Save