metal : ne01 check no longer necessary

2024-08-26 16:24:26 +03:00 · 2024-08-26 16:24:26 +03:00 · e865686c21
commit e865686c21
parent e65fc9b8b2
1 changed files with 1 additions and 1 deletions
--- a/ggml/src/ggml-metal.metal
+++ b/ggml/src/ggml-metal.metal
@ -2144,7 +2144,7 @@ kernel void kernel_flash_attn_ext_f16(
            }

            // scale and apply the mask (assume C = 32)
-            for (short j = 0; j < Q && iq1 + j < ne01; ++j) {
+            for (short j = 0; j < Q; ++j) {
                // mqk = mqk*scale
                ss[j*TF + tiisg] *= scale;