Skip to content

Commit 8252615

Browse files
try CI fix
1 parent 26349e1 commit 8252615

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

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

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1203,10 +1203,11 @@ static bool ggml_metal_supports_op(const struct ggml_backend_metal_device_contex
12031203
case GGML_OP_GROUP_NORM:
12041204
return has_simdgroup_reduction;
12051205
case GGML_OP_RMS_NORM:
1206-
return has_simdgroup_reduction && (op->ne[0] % 4 == 0);
1206+
return has_simdgroup_reduction && (op->ne[0] % 4 == 0 && ggml_is_contiguous_1(src0);
12071207
case GGML_OP_ARGMAX:
1208-
case GGML_OP_NORM:
12091208
return true;
1209+
case GGML_OP_NORM:
1210+
return has_simdgroup_reduction && (op->ne[0] % 4 == 0 && ggml_is_contiguous_1(src0);
12101211
case GGML_OP_ROPE:
12111212
{
12121213
const int mode = ((const int32_t *) op->op_params)[2];

ggml/src/ggml-vulkan/ggml-vulkan.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8182,9 +8182,11 @@ static bool ggml_backend_vk_device_supports_op(ggml_backend_dev_t dev, const ggm
81828182
case GGML_OP_VIEW:
81838183
case GGML_OP_PERMUTE:
81848184
case GGML_OP_TRANSPOSE:
8185+
return true;
81858186
case GGML_OP_NORM:
81868187
case GGML_OP_GROUP_NORM:
81878188
case GGML_OP_RMS_NORM:
8189+
return ggml_is_contiguous(op->src[0]);
81888190
case GGML_OP_ADD:
81898191
case GGML_OP_ACC:
81908192
case GGML_OP_MUL:

0 commit comments

Comments
 (0)