Using static_cast to make more robust.

PaddlePaddle · Oct 27, 2017 · ac5f421 · ac5f421
1 parent 5939a17
commit ac5f421
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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);
     }
   }