Skip to content

Commit 93bf21d

Browse files
committed
fix(metal): add turbo2/3/4 types to FLASH_ATTN_EXT and CPY support checks
1 parent 59798f1 commit 93bf21d

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
@@ -1221,6 +1221,9 @@ bool ggml_metal_device_supports_op(ggml_metal_device_t dev, const struct ggml_te
12211221
case GGML_TYPE_Q4_1:
12221222
case GGML_TYPE_Q5_0:
12231223
case GGML_TYPE_Q5_1:
1224+
case GGML_TYPE_TURBO2_0:
1225+
case GGML_TYPE_TURBO3_0:
1226+
case GGML_TYPE_TURBO4_0:
12241227
break;
12251228
case GGML_TYPE_BF16:
12261229
if (!has_bfloat) {
@@ -1264,6 +1267,9 @@ bool ggml_metal_device_supports_op(ggml_metal_device_t dev, const struct ggml_te
12641267
case GGML_TYPE_Q5_1:
12651268
case GGML_TYPE_IQ4_NL:
12661269
case GGML_TYPE_I32:
1270+
case GGML_TYPE_TURBO2_0:
1271+
case GGML_TYPE_TURBO3_0:
1272+
case GGML_TYPE_TURBO4_0:
12671273
return true;
12681274
default:
12691275
return false;

0 commit comments

Comments
 (0)