Skip to content

Commit

Permalink
Merge pull request PaddlePaddle#34 from mthreads/fix_distribution_bug
Browse files Browse the repository at this point in the history
[MTAI-484] fix(build): repleace murand_uniform with murand_uniform2
  • Loading branch information
caizhi-mt authored and mt-robot committed Aug 13, 2023
2 parents 1f75c99 + 7ed5e9f commit 172dc98
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions paddle/phi/kernels/funcs/distribution_helper.h
Original file line number Diff line number Diff line change
Expand Up @@ -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;
};
Expand Down Expand Up @@ -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;
};
Expand Down

0 comments on commit 172dc98

Please sign in to comment.