fix scale bug

2026-03-13 09:39:19 +01:00 · 2026-03-13 09:39:19 +01:00 · 3ed7d795b5
parent 2688a9657c
commit 3ed7d795b5
1 changed files with 1 additions and 1 deletions
--- a/ggml/src/ggml-metal/ggml-metal.metal
+++ b/ggml/src/ggml-metal/ggml-metal.metal
@ -2871,7 +2871,7 @@ kernel void kernel_l2_norm_impl(
    sumf = shmem_f32[tiisg];
    sumf = simd_sum(sumf);

-    const float scale = 1.0f/sqrt(max(sumf, args.eps));
+    const float scale = 1.0f/max(sqrt(sumf), args.eps);

    for (int i00 = tpitg.x; i00 < args.ne00; i00 += ntg.x) {
        y[i00] = x[i00] * scale;