diff --git a/ggml-cuda/unary.cu b/ggml-cuda/unary.cu index c0ddb8a56..f9e208011 100644 --- a/ggml-cuda/unary.cu +++ b/ggml-cuda/unary.cu @@ -98,7 +98,7 @@ static __global__ void sqrt_f32(const float * x, float * dst, const int k) { if (i >= k) { return; } - dst[i] = sqrt(x[i]); + dst[i] = sqrtf(x[i]); } static void gelu_f32_cuda(const float * x, float * dst, const int k, cudaStream_t stream) {