From 09b7731b8d2e9cc3700b097792be7f1daabc9095 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Fri, 23 Jun 2023 13:10:26 +0200 Subject: [PATCH] Fix unary op. --- kernels/src/unary.cu | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/kernels/src/unary.cu b/kernels/src/unary.cu index faf73bbf..3fe830e1 100644 --- a/kernels/src/unary.cu +++ b/kernels/src/unary.cu @@ -36,12 +36,12 @@ UNARY_OP(__half, gelu_f16, gelu_fwd(x)) #endif UNARY_OP(float, ucopy_f32, x) -UNARY_OP(float, ucopy_f64, x) +UNARY_OP(double, ucopy_f64, x) UNARY_OP(float, uneg_f32, -x) -UNARY_OP(float, uneg_f64, -x) +UNARY_OP(double, uneg_f64, -x) UNARY_OP(float, usqr_f32, x*x) -UNARY_OP(float, usqr_f64, x*x) +UNARY_OP(double, usqr_f64, x*x) UNARY_OP(float, usqrt_f32, sqrtg(x)) -UNARY_OP(float, usqrt_f64, sqrtg(x)) +UNARY_OP(double, usqrt_f64, sqrtg(x)) UNARY_OP(float, gelu_f32, gelu_fwd(x)) -UNARY_OP(float, gelu_f64, gelu_fwd(x)) +UNARY_OP(double, gelu_f64, gelu_fwd(x))