fix compile bug of windows cuda11.5 (PaddlePaddle#41433)

douch · Apr 10, 2022 · 8769a9d · 8769a9d
1 parent 24739f9
commit 8769a9d
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/paddle/phi/kernels/funcs/activation_functor.h b/paddle/phi/kernels/funcs/activation_functor.h
@@ -1878,12 +1878,17 @@ struct CudaCosGradFunctor : public BaseActivationFunctor<T> {
 
 template <typename T>
 struct CudaExpFunctor : public BaseActivationFunctor<T> {
-  using MPType = typename phi::dtype::MPTypeTrait<T>::Type;
+  // exp(x) = expf(x)
+  __device__ __forceinline__ T operator()(const T x) const {
+    return static_cast<T>(expf(static_cast<float>(x)));
+  }
+};
 
+template <>
+struct CudaExpFunctor<double> : public BaseActivationFunctor<double> {
   // exp(x) = exp(x)
-  __device__ __forceinline__ T operator()(const T arg_x) const {
-    MPType x = static_cast<MPType>(arg_x);
-    return static_cast<T>(exp(x));
+  __device__ __forceinline__ double operator()(const double x) const {
+    return exp(x);
   }
 };