Skip to content

Commit 0e5e019

Browse files
authored
add metal check
1 parent c91285a commit 0e5e019

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -683,9 +683,11 @@ bool ggml_metal_device_supports_op(ggml_metal_device_t dev, const struct ggml_te
683683
(ggml_get_op_params_i32(op, 4) == 0) && (ggml_get_op_params_i32(op, 6) == 0);
684684
case GGML_OP_PAD_REFLECT_1D:
685685
case GGML_OP_TIMESTEP_EMBEDDING:
686-
case GGML_OP_ARGSORT:
687686
case GGML_OP_LEAKY_RELU:
688687
return op->src[0]->type == GGML_TYPE_F32;
688+
case GGML_OP_ARGSORT:
689+
// TODO: Support arbitrary column width
690+
return op->src[0]->ne[0] <= 1024;
689691
case GGML_OP_ARANGE:
690692
return true;
691693
case GGML_OP_FLASH_ATTN_EXT:

0 commit comments

Comments
 (0)