From ac5f42184f56029631a29e1c62b1b527c4cd0bfc Mon Sep 17 00:00:00 2001
From: yangyaming <mxscmxsc@gmail.com>
Date: Fri, 27 Oct 2017 11:54:50 +0800
Subject: [PATCH] Using static_cast to make more robust.

---
 paddle/operators/huber_loss_op.h | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle/operators/huber_loss_op.h b/paddle/operators/huber_loss_op.h
index d8a2da52f..4e7bc5543 100644
--- a/paddle/operators/huber_loss_op.h
+++ b/paddle/operators/huber_loss_op.h
@@ -32,9 +32,9 @@ struct HuberLossForward {
   HOSTDEVICE T operator()(const T& val) const {
     T abs_val = std::abs(val);
     if (abs_val <= delta) {
-      return 0.5 * val * val;
+      return static_cast<T>(0.5) * val * val;
     } else {
-      return delta * (abs_val - 0.5 * delta);
+      return delta * (abs_val - static_cast<T>(0.5) * delta);
     }
   }
 
-- 
GitLab