Skip to content

Commit f4010c5

Browse files
committed
cont : fix supports_op for ARGMAX
1 parent 1d31812 commit f4010c5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1982,7 +1982,7 @@ static bool ggml_metal_supports_op(const struct ggml_backend_metal_device_contex
19821982
case GGML_OP_L2_NORM:
19831983
return has_simdgroup_reduction && (op->ne[0] % 4 == 0 && ggml_is_contiguous_1(op->src[0]));
19841984
case GGML_OP_ARGMAX:
1985-
return true;
1985+
return has_simdgroup_reduction;
19861986
case GGML_OP_NORM:
19871987
return has_simdgroup_reduction && (op->ne[0] % 4 == 0 && ggml_is_contiguous_1(op->src[0]));
19881988
case GGML_OP_ROPE:

0 commit comments

Comments
 (0)