diff --git a/paddle/fluid/operators/detection/yolo_box_op.cu b/paddle/fluid/operators/detection/yolo_box_op.cu index 7d223e84a8be247508d3a611115f977198e35293..5a882958e66a79507e053a96b15be8cbbcc83164 100644 --- a/paddle/fluid/operators/detection/yolo_box_op.cu +++ b/paddle/fluid/operators/detection/yolo_box_op.cu @@ -84,15 +84,15 @@ class YoloBoxOpCUDAKernel : public framework::OpKernel { int input_size = downsample_ratio * h; auto& dev_ctx = ctx.cuda_device_context(); - auto& allocator = - platform::DeviceTemporaryAllocator::Instance().Get(dev_ctx); + auto& allocator = + platform::DeviceTemporaryAllocator::Instance().Get(dev_ctx); int bytes = sizeof(int) * anchors.size(); auto anchors_ptr = allocator.Allocate(sizeof(int) * anchors.size()); int* anchors_data = reinterpret_cast(anchors_ptr->ptr()); const auto gplace = boost::get(ctx.GetPlace()); const auto cplace = platform::CPUPlace(); memory::Copy(gplace, anchors_data, cplace, anchors.data(), bytes, - dev_ctx.stream()); + dev_ctx.stream()); const T* input_data = input->data(); const int* imgsize_data = img_size->data();