diff --git a/oneflow/user/kernels/upsample_kernel.h b/oneflow/user/kernels/upsample_kernel.h index d783413a57e..d62a659e319 100644 --- a/oneflow/user/kernels/upsample_kernel.h +++ b/oneflow/user/kernels/upsample_kernel.h @@ -138,7 +138,7 @@ OF_DEVICE_FUNC T cubic_convolution2(const T x, const T A) { template OF_DEVICE_FUNC void get_cubic_upsample_coefficients(T coeffs[4], const T t) { - T A = -0.75; + float A = -0.75; T x1 = t; coeffs[0] = cubic_convolution2(x1 + 1.0, A);