Merge pull request PaddlePaddle#34 from mthreads/fix_distribution_bug

[MTAI-484] fix(build): repleace murand_uniform with murand_uniform2
hanhaowen-mt · Aug 13, 2023 · 172dc98 · 172dc98
2 parents 1f75c99 + 7ed5e9f
commit 172dc98
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/phi/kernels/funcs/distribution_helper.h b/paddle/phi/kernels/funcs/distribution_helper.h
@@ -210,7 +210,7 @@ template <>
 struct uniform_distribution<double> {
   __device__ inline double2 operator()(
       murand_state_philox4x32_10 *state) const {
-    return murand_uniform_double2(state);
+    return murand_uniform2_double(state);
   }
   static constexpr int kReturnsCount = 2;
 };
@@ -248,7 +248,7 @@ template <>
 struct normal_distribution<double> {
   __device__ inline double2 operator()(
       murand_state_philox4x32_10 *state) const {
-    return murand_normal_double2(state);
+    return murand_normal2_double(state);
   }
   static constexpr int kReturnsCount = 2;
 };