From ddd3c15174286b8401049d857aed13517a65eafd Mon Sep 17 00:00:00 2001
From: baiyfbupt <baiyfbupt@gmail.com>
Date: Wed, 16 May 2018 17:10:34 +0000
Subject: [PATCH] bug fix

---
 paddle/fluid/operators/roi_pool_op.cu | 24 ++++++++++++------------
 1 file changed, 12 insertions(+), 12 deletions(-)

diff --git a/paddle/fluid/operators/roi_pool_op.cu b/paddle/fluid/operators/roi_pool_op.cu
index 265dbb3a801..a699d21542a 100644
--- a/paddle/fluid/operators/roi_pool_op.cu
+++ b/paddle/fluid/operators/roi_pool_op.cu
@@ -53,18 +53,18 @@ __global__ void GPUROIPoolForward(
     int roi_width = max(roi_end_w - roi_start_w + 1, 1);
     int roi_height = max(roi_end_h - roi_start_h + 1, 1);
 
-    int hstart =
-        static_cast<int>(floor(static_cast<T>(ph) * static_cast<T>(roi_height) /
-                               static_cast<T>(pooled_height)));
-    int wstart =
-        static_cast<int>(floor(static_cast<T>(pw) * static_cast<T>(roi_width) /
-                               static_cast<T>(pooled_width)));
-    int hend = static_cast<int>(ceil(static_cast<T>(ph + 1) *
-                                     static_cast<T>(roi_height) /
-                                     static_cast<T>(pooled_height)));
-    int wend = static_cast<int>(ceil(static_cast<T>(pw + 1) *
-                                     static_cast<T>(roi_width) /
-                                     static_cast<T>(pooled_width)));
+    int hstart = static_cast<int>(floor(static_cast<double>(ph) *
+                                        static_cast<double>(roi_height) /
+                                        static_cast<double>(pooled_height)));
+    int wstart = static_cast<int>(floor(static_cast<double>(pw) *
+                                        static_cast<double>(roi_width) /
+                                        static_cast<double>(pooled_width)));
+    int hend = static_cast<int>(ceil(static_cast<double>(ph + 1) *
+                                     static_cast<double>(roi_height) /
+                                     static_cast<double>(pooled_height)));
+    int wend = static_cast<int>(ceil(static_cast<double>(pw + 1) *
+                                     static_cast<double>(roi_width) /
+                                     static_cast<double>(pooled_width)));
 
     hstart = min(max(hstart + roi_start_h, 0), height);
     hend = min(max(hend + roi_start_h, 0), height);
-- 
GitLab