未验证 提交 2db77be4 编写于 作者: D Double_V 提交者: GitHub

fix wrong data type, test=develop (#28203)

上级 efe6e284
...@@ -62,7 +62,7 @@ class XPUROIAlignOpKernel : public framework::OpKernel<T> { ...@@ -62,7 +62,7 @@ class XPUROIAlignOpKernel : public framework::OpKernel<T> {
} else { } else {
auto _rois_lod = rois->lod().back(); auto _rois_lod = rois->lod().back();
rois_batch_size = _rois_lod.size() - 1; rois_batch_size = _rois_lod.size() - 1;
for (int n = 0; n < _rois_lod.size(); ++n) { for (int n = 0; n < static_cast<int>(_rois_lod.size()); ++n) {
rois_lod[n] = _rois_lod[n]; rois_lod[n] = _rois_lod[n];
} }
PADDLE_ENFORCE_EQ( PADDLE_ENFORCE_EQ(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册