Skip to content

Commit 1191c82

Browse files
committed
https://github.com/ggerganov/llama.cpp/commit/d774ab3acc4fee41fbed6dbfc192b57d5f79f34b#diff-4045ccc2edf36af53c4a28fca4a6a24f628b3c9bee7ccb43b06e55dc5e5e199e
1 parent 5d09cb2 commit 1191c82

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
@@ -1194,7 +1194,7 @@ static bool ggml_metal_supports_op(const struct ggml_backend_metal_device_contex
11941194
case GGML_OP_SUM_ROWS:
11951195
case GGML_OP_SOFT_MAX:
11961196
case GGML_OP_GROUP_NORM:
1197-
return has_simdgroup_reduction;
1197+
return has_simdgroup_reduction && (op->ne[0] % 4 == 0 && ggml_is_contiguous_1(op->src[0]));
11981198
case GGML_OP_RMS_NORM:
11991199
return has_simdgroup_reduction && (op->ne[0] % 4 == 0);
12001200
case GGML_OP_ARGMAX:

0 commit comments

Comments
 (0)