Skip to content

Commit 6d0370d

Browse files
committed
fix warning
1 parent 0d75191 commit 6d0370d

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

paddle/fluid/operators/cross_entropy_op.h

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -86,10 +86,10 @@ class XeGradFunctor {
8686
auto x_is_true_offset = sample_id * num_classes_ + label_[sample_id];
8787
for (size_t x_offset = sample_id * num_classes_;
8888
x_offset < (sample_id + 1) * num_classes_; ++x_offset) {
89-
dx_[x_offset] =
90-
(x_offset != x_is_true_offset || label_[sample_id] == ignore_index_)
91-
? static_cast<T>(0)
92-
: -dy_[sample_id] / x_[x_offset];
89+
dx_[x_offset] = (x_offset != x_is_true_offset ||
90+
label_[sample_id] == static_cast<int64_t>(ignore_index_))
91+
? static_cast<T>(0)
92+
: -dy_[sample_id] / x_[x_offset];
9393
}
9494
}
9595

0 commit comments

Comments
 (0)