We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1d31812 commit f4010c5Copy full SHA for f4010c5
ggml/src/ggml-metal/ggml-metal.m
@@ -1982,7 +1982,7 @@ static bool ggml_metal_supports_op(const struct ggml_backend_metal_device_contex
1982
case GGML_OP_L2_NORM:
1983
return has_simdgroup_reduction && (op->ne[0] % 4 == 0 && ggml_is_contiguous_1(op->src[0]));
1984
case GGML_OP_ARGMAX:
1985
- return true;
+ return has_simdgroup_reduction;
1986
case GGML_OP_NORM:
1987
1988
case GGML_OP_ROPE:
0 commit comments