metal : fix -INF detection logic

ggerganov · ggerganov · commit c619c088ce80 · 2025-10-02T08:38:15.000+03:00
diff --git a/ggml/src/ggml-metal/ggml-metal.metal b/ggml/src/ggml-metal/ggml-metal.metal
@@ -4559,7 +4559,7 @@ kernel void kernel_flash_attn_ext_blk(
 
         FOR_UNROLL (short j = 0; j < Q; ++j) {
             FOR_UNROLL (short ii = 0; ii < C/NW; ++ii) {
-                m = max(m, *mask_src);
+                m = max(m, mask_src[ii*NW]);
             }
 
             mask_src += args.nb31/2;

Original file line number	Diff line number	Diff line change
`@@ -4559,7 +4559,7 @@ kernel void kernel_flash_attn_ext_blk(`
`4559`	`4559`
`4560`	`4560`	`FOR_UNROLL (short j = 0; j < Q; ++j) {`
`4561`	`4561`	`FOR_UNROLL (short ii = 0; ii < C/NW; ++ii) {`
`4562`		`- m = max(m, *mask_src);`
	`4562`	`+ m = max(m, mask_src[ii*NW]);`
`4563`	`4563`	`}`
`4564`	`4564`
`4565`	`4565`	`mask_src += args.nb31/2;`