fix cuda yolobox kernel of the input type, test=develop (#2740)

dafbc21c · juncaipeng · GitHub · e530d2f3 · dafbc21c
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

lite/kernels/cuda/yolo_box_compute.cu lite/kernels/cuda/yolo_box_compute.cu +1 -1

未找到文件。
--- a/lite/kernels/cuda/yolo_box_compute.cu
+++ b/lite/kernels/cuda/yolo_box_compute.cu
@@ -233,7 +233,7 @@ REGISTER_LITE_KERNEL(yolo_box,
                                      DATALAYOUT(kNCHW))})
    .BindInput("ImgSize",
               {LiteType::GetTensorTy(TARGET(kCUDA),
-                                      PRECISION(kFloat),
+                                      PRECISION(kInt32),
                                      DATALAYOUT(kNCHW))})
    .BindOutput("Boxes",
                {LiteType::GetTensorTy(TARGET(kCUDA),