Skip to content

Commit

Permalink
typo pre_dim
Browse files Browse the repository at this point in the history
  • Loading branch information
thunder95 committed May 12, 2022
1 parent bcfb015 commit 718fcdb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion paddle/phi/kernels/cpu/nanmedian_kernel.cc
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ void CalcMedianFunc(const Context& dev_ctx,
bool is_ori_odd = stride & 1;
if (should_ignore_nan) {
for (i = 0; i < pre_dim; i++) {
offset = i * pre_dim;
offset = i * sort_k;
if (nan_counts[i] == stride) {
m_ptr[i * 2] = -1;
m_ptr[i * 2 + 1] = -1;
Expand Down

0 comments on commit 718fcdb

Please sign in to comment.