From c929963d741d782d819e412006d45dec95ece7bc Mon Sep 17 00:00:00 2001
From: Weixing Zhang <weixingzhang@users.noreply.github.com>
Date: Fri, 24 Apr 2020 00:49:37 -0700
Subject: [PATCH] type cast for ratio is not necessary for dropout (#3682)

Co-authored-by: Weixing Zhang <wezhan@microsoft.com>
---
 orttraining/orttraining/training_ops/cuda/nn/dropout.cc | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/orttraining/orttraining/training_ops/cuda/nn/dropout.cc b/orttraining/orttraining/training_ops/cuda/nn/dropout.cc
index bced973f88..44e39f8af3 100644
--- a/orttraining/orttraining/training_ops/cuda/nn/dropout.cc
+++ b/orttraining/orttraining/training_ops/cuda/nn/dropout.cc
@@ -37,7 +37,6 @@ REGISTER_KERNEL_TYPED(Dropout, kOnnxDomain, 12, double, double, 1)
 template <typename T1, typename T2>
 Status Dropout<T1, T2>::ComputeInternal(OpKernelContext* context) const {
   typedef typename ToCudaType<T1>::MappedType CudaT;
-  typedef typename ToCudaType<T2>::MappedType CudaT2;
 
   //Get X_data
   const Tensor* X = context->Input<Tensor>(0);
@@ -68,7 +67,7 @@ Status Dropout<T1, T2>::ComputeInternal(OpKernelContext* context) const {
                 "T2 must be float16 or float or double");
 
   if (ratio) {
-    ratio_data = static_cast<float>(*reinterpret_cast<const CudaT2*>(ratio->template Data<T2>()));
+    ratio_data = static_cast<float>(*(ratio->template Data<T2>()));
   } else {
     ratio_data = default_ratio_;
   }
@@ -112,7 +111,7 @@ Status DropoutGrad<T1, T2>::ComputeInternal(OpKernelContext* context) const {
                 "T2 must be float16 or float or double");
 
   if (ratio) {
-    ratio_data = static_cast<float>(*reinterpret_cast<const T2*>(ratio->template Data<T2>()));
+    ratio_data = static_cast<float>(*(ratio->template Data<T2>()));
   } else {
     ratio_data = default_ratio_;
   }