Roi align kl (#28128)

* add rois_num params for roi_align_xpu op, test=kunlun * fix xpu roi_align_op warning,test=kunlun
5 years ago · 39745d448c
parent 172e76a852
commit 39745d448c
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/roi_align_op_xpu.cc
+++ b/paddle/fluid/operators/roi_align_op_xpu.cc
@ -62,7 +62,7 @@ class XPUROIAlignOpKernel : public framework::OpKernel<T> {
    } else {
      auto _rois_lod = rois->lod().back();
      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];
      }
      PADDLE_ENFORCE_EQ(