未验证 提交 dcf30692 编写于 作者: W wanghuancoder 提交者: GitHub

fix security bug (#55865)

* fix security bug
上级 7c9b1ab6
......@@ -123,8 +123,18 @@ class SearchSortedFunctor {
const phi::DDim& val_dims = value_->dims();
bool is_1d_boundaries = seq_dims.size() == 1;
int64_t val_size = val_dims[val_dims.size() - 1];
int64_t seq_size = seq_dims[seq_dims.size() - 1];
int64_t val_size = 0;
int64_t seq_size = 0;
if (val_dims.size()) {
val_size = val_dims[val_dims.size() - 1];
} else {
val_size = 1;
}
if (seq_dims.size()) {
seq_size = seq_dims[seq_dims.size() - 1];
} else {
seq_size = 1;
}
funcs::ForRange<Context> for_range(context_, value_->numel());
GpuAndCpuSearchSortedCompute<T1, T2, OutType>
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册