From dafbc21cd8d43c7581b208395febec8a2449e3a3 Mon Sep 17 00:00:00 2001
From: juncaipeng <52520497+juncaipeng@users.noreply.github.com>
Date: Thu, 9 Jan 2020 10:22:39 +0800
Subject: [PATCH] fix cuda yolobox kernel of the input type, test=develop
 (#2740)

---
 lite/kernels/cuda/yolo_box_compute.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lite/kernels/cuda/yolo_box_compute.cu b/lite/kernels/cuda/yolo_box_compute.cu
index 0a00c06cbf..6b4b2875f3 100644
--- 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),
-- 
GitLab