Fix cuda warning

Signed-off-by: Molly Sophia <[email protected]>
ggerganov · Dec 28, 2024 · 060e57a · 060e57a
1 parent 74af642
commit 060e57a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ggml/src/ggml-cuda/gla.cu b/ggml/src/ggml-cuda/gla.cu
@@ -73,7 +73,8 @@ void ggml_cuda_op_gated_linear_attn(ggml_backend_cuda_context & ctx, ggml_tensor
     const int64_t C = dst->ne[0];
     const int64_t H = dst->src[0]->ne[1];
 
-    const float scale = ((const float*)(dst->op_params))[0];
+    float scale;
+    memcpy(&scale, (float*)dst->op_params, sizeof(float));
 
     float * dst_d = (float *)dst->data;