compile fix

5 years ago · 2500278784
parent 0acd6990d8
commit 2500278784
1 changed files with 2 additions and 2 deletions
--- 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);