File tree 1 file changed +2
-2
lines changed
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -1204,13 +1204,13 @@ static bool ggml_metal_supports_op(const struct ggml_backend_metal_device_contex
1204
1204
case GGML_OP_SUM_ROWS:
1205
1205
case GGML_OP_SOFT_MAX:
1206
1206
case GGML_OP_GROUP_NORM:
1207
- return has_simdgroup_reduction;
1207
+ return has_simdgroup_reduction && ggml_is_contiguous(op->src[0]) ;
1208
1208
case GGML_OP_RMS_NORM:
1209
1209
return has_simdgroup_reduction && (op->ne[0] % 4 == 0 && ggml_is_contiguous_1(op->src[0]));
1210
1210
case GGML_OP_ARGMAX:
1211
1211
return true;
1212
1212
case GGML_OP_NORM:
1213
- return has_simdgroup_reduction && ggml_is_contiguous (op->src[0]);
1213
+ return has_simdgroup_reduction && (op->ne[0] % 4 == 0 && ggml_is_contiguous_1(op-> src[0]) );
1214
1214
case GGML_OP_ROPE:
1215
1215
{
1216
1216
const int mode = ((const int32_t *) op->op_params)[2];
You can’t perform that action at this time.
0 commit comments