Compile fix

This commit is contained in:
AlexeyAB 2020-06-15 22:14:01 +03:00
parent d724306ee3
commit e08a818d5f
1 changed files with 2 additions and 2 deletions

View File

@ -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);