metal : fix -INF detection logic

ggerganov · ggerganov · commit f478afab9e97 · 2025-10-07T10:07:40.000+03:00
diff --git a/ggml/src/ggml-metal/ggml-metal.metal b/ggml/src/ggml-metal/ggml-metal.metal
@@ -4459,7 +4459,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
`@@ -4459,7 +4459,7 @@ kernel void kernel_flash_attn_ext_blk(`
`4459`	`4459`
`4460`	`4460`	`FOR_UNROLL (short j = 0; j < Q; ++j) {`
`4461`	`4461`	`FOR_UNROLL (short ii = 0; ii < C/NW; ++ii) {`
`4462`		`- m = max(m, *mask_src);`
	`4462`	`+ m = max(m, mask_src[ii*NW]);`
`4463`	`4463`	`}`
`4464`	`4464`
`4465`	`4465`	`mask_src += args.nb31/2;`