提交 8e0b9496 编写于 作者: D Dang Qingqing

Fix unit test

test=develop
上级 593ad763
...@@ -418,7 +418,7 @@ class CUDAGenerateProposalsKernel : public framework::OpKernel<T> { ...@@ -418,7 +418,7 @@ class CUDAGenerateProposalsKernel : public framework::OpKernel<T> {
T *rpn_rois_data = rpn_rois->data<T>(); T *rpn_rois_data = rpn_rois->data<T>();
T *rpn_roi_probs_data = rpn_roi_probs->data<T>(); T *rpn_roi_probs_data = rpn_roi_probs->data<T>();
auto &place = boost::get<platform::CUDAPlace>(dev_ctx.GetPlace()); auto place = boost::get<platform::CUDAPlace>(dev_ctx.GetPlace());
int64_t num_proposals = 0; int64_t num_proposals = 0;
std::vector<size_t> offset(1, 0); std::vector<size_t> offset(1, 0);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册