Skip to content

Commit 0d0d828

Browse files
tomwolfeclaude
authored andcommitted
fix(metal): add turbo2/3/4 types to FLASH_ATTN_EXT and CPY support checks
1 parent fe1e09a commit 0d0d828

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

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

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1277,6 +1277,9 @@ bool ggml_metal_device_supports_op(ggml_metal_device_t dev, const struct ggml_te
12771277
case GGML_TYPE_Q4_1:
12781278
case GGML_TYPE_Q5_0:
12791279
case GGML_TYPE_Q5_1:
1280+
case GGML_TYPE_TURBO2_0:
1281+
case GGML_TYPE_TURBO3_0:
1282+
case GGML_TYPE_TURBO4_0:
12801283
break;
12811284
case GGML_TYPE_BF16:
12821285
if (!has_bfloat) {
@@ -1320,6 +1323,9 @@ bool ggml_metal_device_supports_op(ggml_metal_device_t dev, const struct ggml_te
13201323
case GGML_TYPE_Q5_1:
13211324
case GGML_TYPE_IQ4_NL:
13221325
case GGML_TYPE_I32:
1326+
case GGML_TYPE_TURBO2_0:
1327+
case GGML_TYPE_TURBO3_0:
1328+
case GGML_TYPE_TURBO4_0:
13231329
return true;
13241330
default:
13251331
return false;

0 commit comments

Comments
 (0)