fix gcc-7 / cuda10 compilation

soumith · soumith · commit 1031028f3b04 · 2019-05-05T13:58:27.000Z
diff --git a/cuda/lltm_cuda_kernel.cu b/cuda/lltm_cuda_kernel.cu
@@ -25,7 +25,7 @@ __device__ __forceinline__ scalar_t d_tanh(scalar_t z) {
 
 template <typename scalar_t>
 __device__ __forceinline__ scalar_t elu(scalar_t z, scalar_t alpha = 1.0) {
-  return fmax(0.0, z) + fmin(0.0, alpha * (exp(z) - 1.0));
+  return fmaxf(0.0, z) + fminf(0.0, alpha * (exp(z) - 1.0));
 }
 
 template <typename scalar_t>

Original file line number	Diff line number	Diff line change
`@@ -25,7 +25,7 @@ __device__ __forceinline__ scalar_t d_tanh(scalar_t z) {`
`25`	`25`
`26`	`26`	`template <typename scalar_t>`
`27`	`27`	`__device__ __forceinline__ scalar_t elu(scalar_t z, scalar_t alpha = 1.0) {`
`28`		`- return fmax(0.0, z) + fmin(0.0, alpha * (exp(z) - 1.0));`
	`28`	`+ return fmaxf(0.0, z) + fminf(0.0, alpha * (exp(z) - 1.0));`
`29`	`29`	`}`
`30`	`30`
`31`	`31`	`template <typename scalar_t>`