compile fix

AlexeyAB · May 12, 2020 · 2500278 · 2500278
1 parent 0acd699
commit 2500278
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/activation_kernels.cu b/src/activation_kernels.cu
@@ -233,11 +233,11 @@ __device__ float mish_njuffa(float x)
 
 __device__ float mish_yashas(float x)
 {
-    auto e = __expf(x);
+    float e = __expf(x);
     if (x <= -18.0f)
         return x * e;
 
-    auto n = e * e + 2 * e;
+    float n = e * e + 2 * e;
     if (x <= -5.0f)
         return x * __fdividef(n, n + 2);