Skip to content

Commit 2248b56

Browse files
Merge pull request #291 from menloresearch/update-dev-from-master-2025-10-14-06-52
Sync master with upstream release b6756
2 parents fc090ec + bc07349 commit 2248b56

File tree

5 files changed

+125
-63
lines changed

5 files changed

+125
-63
lines changed

ggml/src/ggml-metal/ggml-metal-device.m

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -693,7 +693,8 @@ bool ggml_metal_device_supports_op(ggml_metal_device_t dev, const struct ggml_te
693693
return true;
694694
case GGML_OP_FLASH_ATTN_EXT:
695695
// for new head sizes, add checks here
696-
if (op->src[0]->ne[0] != 40 &&
696+
if (op->src[0]->ne[0] != 32 &&
697+
op->src[0]->ne[0] != 40 &&
697698
op->src[0]->ne[0] != 64 &&
698699
op->src[0]->ne[0] != 80 &&
699700
op->src[0]->ne[0] != 96 &&

0 commit comments

Comments
 (0)