Skip to content

Commit 88e73e8

Browse files
committed
metal : fix QK unroll condition
1 parent 347c113 commit 88e73e8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml/src/ggml-metal/ggml-metal.metal

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4791,7 +4791,7 @@ void kernel_flash_attn_ext_impl(
47914791
for (short cc = 0; cc < NC; ++cc) {
47924792
qk8x8_t mqk = make_filled_simdgroup_matrix<qk_t, 8>((qk_t) 0.0f);
47934793

4794-
if (DK8 % 16 != 0) {
4794+
if (DK % 16 != 0) {
47954795
k8x8_t mk;
47964796
q8x8_t mq;
47974797

0 commit comments

Comments
 (0)