diff --git a/paddle/phi/kernels/funcs/distribution_helper.h b/paddle/phi/kernels/funcs/distribution_helper.h index 484f077583bef..4705370f71f7c 100644 --- a/paddle/phi/kernels/funcs/distribution_helper.h +++ b/paddle/phi/kernels/funcs/distribution_helper.h @@ -210,7 +210,7 @@ template <> struct uniform_distribution { __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 { __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; };