Compile fix

AlexeyAB · Jun 15, 2020 · e08a818 · e08a818
1 parent d724306
commit e08a818
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/activation_kernels.cu b/src/activation_kernels.cu
@@ -247,8 +247,8 @@ __device__ float mish_yashas(float x)
 
 __device__ float mish_yashas2(float x)
 {
-    auto e = __expf(x);
-    auto n = e * e + 2 * e;
+    float e = __expf(x);
+    float n = e * e + 2 * e;
     if (x <= -0.6f)
         return x * __fdividef(n, n + 2);